Update app.py

app.py

@@ -8,22 +8,23 @@ from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDisc
 
 model_id = "timbrooks/instruct-pix2pix"
 pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
-pipe.to("cpu")
+pipe.to("cuda")
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
-def download_image(img):
-    image = PIL.Image.open(requests.get(img, stream=True).raw)
-    image = PIL.ImageOps.exif_transpose(image)
-    image = image.convert("RGB")
-    return image
-image = download_image("https://raw.githubusercontent.com/timothybrooks/instruct-pix2pix/main/imgs/example.jpg")
+
 
 def greet(name):
     prompt = "add tshirt"
+    def download_image(img):
+        image = PIL.Image.open(requests.get(img, stream=True).raw)
+        image = PIL.ImageOps.exif_transpose(image)
+        image = image.convert("RGB")
+        return image
+    image = download_image("https://raw.githubusercontent.com/timothybrooks/instruct-pix2pix/main/imgs/example.jpg")
     images = pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images
       
     
-    image.save(images[0]+".png")
+    #image.save(images[0]+".png")
     return "done"
 
 iface = gr.Interface(fn=greet, inputs="text", outputs="text")