Update app.py

app.py

@@ -34,7 +34,7 @@ def predict(sketch, description):
     # pipe.enable_xformers_memory_efficient_attention()
 
     # Move pipeline to GPU
-    pipe = pipe.to("cuda")
+    # pipe = pipe.to("cuda")
 
     result = pipe(description, image, num_inference_steps=10).images[0]