Update app.py

app.py

@@ -16,6 +16,8 @@ image_pipeline.set_ip_adapter_scale(0.6)
 
 @spaces.GPU(enable_queue=True)
 def text_to_image(ip, prompt, neg_prompt, width, height, ip_scale, strength, guidance, steps):
+    text_pipeline.to("cuda")
+    
     ip.thumbnail((1024, 1024))
     
     text_pipeline.set_ip_adapter_scale(ip_scale)
@@ -35,6 +37,8 @@ def text_to_image(ip, prompt, neg_prompt, width, height, ip_scale, strength, gui
 
 @spaces.GPU(enable_queue=True)
 def image_to_image(ip, image, prompt, neg_prompt, width, height, ip_scale, strength, guidance, steps):
+    image_pipeline.to("cuda")
+    
     ip.thumbnail((1024, 1024))
     image.thumbnail((1024, 1024))