Update app.py

app.py

@@ -5,7 +5,11 @@ import requests
 import uuid
 import io 
 import base64
-import threading
+
+import torch
+from diffusers import AutoPipelineForImage2Image
+from diffusers.utils import make_image_grid, load_image
+
 
 loaded_model=[]
 for i,model in enumerate(models):
@@ -16,7 +20,33 @@ for i,model in enumerate(models):
         pass
 print (loaded_model)
 
+
 def run_dif(out_prompt,model_drop,cnt):
+    out_box=[]
+
+    pipeline = AutoPipelineForImage2Image.from_pretrained(
+        "runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
+    )
+    pipeline.enable_model_cpu_offload()
+    # remove following line if xFormers is not installed or you have PyTorch 2.0 or higher installed
+    pipeline.enable_xformers_memory_efficient_attention()
+    
+    # prepare image
+    url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/img2img-init.png"
+    init_image = load_image(url)
+    
+    prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
+    
+    # pass prompt and image to pipeline
+    image = pipeline(prompt, image=init_image, strength=0.8).images[0]
+    #make_image_grid([init_image, image], rows=1, cols=2)
+    out_box.append(image)
+    return out_box,""
+
+
+
+
+def run_dif_old(out_prompt,model_drop,cnt):
     p_seed=""
     out_box=[]
     out_html=""