Update app.py

app.py

@@ -6,6 +6,25 @@ import time
 import os
 import torch
 import spaces
+from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
+from diffusers.utils import load_image
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+model_repo_id = "timbrooks/instruct-pix2pix"
+
+if torch.cuda.is_available():
+    torch_dtype = torch.float16
+else:
+    torch_dtype = torch.float32
+
+pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(
+    model_repo_id, 
+    torch_dtype=torch_dtype, 
+    safety_checker=None
+)
+pipe = pipe.to(device)
+pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+
 @spaces.GPU
 def greet(name):
     return "Hello there {}".format(name)