Update app.py

app.py

@@ -14,6 +14,7 @@ from transformers.generation import (
 )
 import torch
 from emu3.mllm.processing_emu3 import Emu3Processor
+import spaces
 
 # Model paths
 EMU_GEN_HUB = "BAAI/Emu3-Gen"
@@ -59,6 +60,7 @@ chat_processor = Emu3Processor(
     chat_image_processor, chat_image_tokenizer, chat_tokenizer
 )
 
+@spaces.GPU(duration=120)
 def generate_image(prompt):
     POSITIVE_PROMPT = " masterpiece, film grained, best quality."
     NEGATIVE_PROMPT = (