Update app.py

app.py

@@ -197,7 +197,7 @@ def calc_timeout_generic(*args, **kwargs):
         return 60
 
 
-@spaces.GPU(duration=calc_timeout_generic)
+@spaces.GPU(duration=calc_timeout_generic, size="xlarge")
 def generate_image(model_name: str, text: str, image: Image.Image,
                    max_new_tokens: int = 1024, temperature: float = 0.6,
                    top_p: float = 0.9, top_k: int = 50,