Update app.py

app.py

@@ -29,7 +29,7 @@ def load_models():
         torch_dtype=torch.bfloat16
     )
     device = "cuda" if torch.cuda.is_available() else "cpu"
-    pipe.to(device, torch_dtype)
+    # pipe.to(device, torch_dtype)
     
     return model, tokenizer, pipe
 
@@ -90,7 +90,7 @@ def generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale):
     
     return image
 
-@spaces.GPU(duration=100)
+# @spaces.GPU(duration=100)
 def run_pipeline(image_caption, text_caption, seed, num_inference_steps, guidance_scale):
     """Run the complete pipeline from captions to final image"""
     combined_caption, enhanced_caption = generate_enhanced_caption(image_caption, text_caption)