Update app.py

app.py

@@ -6,7 +6,7 @@ import random
 import gradio as gr
 import torch
 from PIL import Image, ImageOps
-from diffusers import StableDiffusionInstructPix2PixPipeline
+from diffusers import StableDiffusionPipeline
 
 
 help_text = """
@@ -18,11 +18,11 @@ example_instructions = [
     "A river"
 ]
 
-model_id = "models/dimentox/heightmapstyle"
+model_id = "dimentox/heightmapstyle"
 
 
 def main():
-    pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None).to("cuda")
+    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None).to("cuda")
     #example_image = Image.open("imgs/example.jpg").convert("RGB")
 
     def load_example(