add img2img diff

app.py

@@ -1,3 +1,18 @@
 import gradio as gr
+import torch
 
-gr.Interface.load("models/shanetx/stable-diffusion-wtx-app").launch()
+from diffusers import StableDiffusionImg2ImgPipeline
+
+# load the pipeline
+device = "cpu"
+model_id_or_path = "runwayml/stable-diffusion-v1-5"
+# model_id_or_path = "shanetx/stable-diffusion-wtx-app"
+img2img_pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float16)
+img2img_pipe = img2img_pipe.to(device)
+
+
+def img2img_diff(prompt, img):
+    return img2img_pipe(prompt=prompt, image=img, strength=0.75, guidance_scale=7.5).images[0]
+
+
+gr.Interface(fn=img2img_diff, inputs=['text', 'image'], outputs='image')

requirements.txt

@@ -0,0 +1,4 @@
+torch
+diffusers
+transformers
+accelerate