fix more

app.py

@@ -11,6 +11,16 @@ from share_btn import community_icon_html, loading_icon_html, share_js
 word_list_dataset = load_dataset("Gustavosta/Stable-Diffusion-Prompts")
 word_list = word_list_dataset["train"]['Prompt']
 
+
+from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
+import torch
+
+model_id = "stabilityai/stable-diffusion-2-1-base"
+
+scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
+pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
+
+
 is_gpu_busy = False
 def infer(prompt, negative, scale):
     global is_gpu_busy
@@ -19,12 +29,17 @@ def infer(prompt, negative, scale):
             raise gr.Error("Unsafe content found. Please try again with different prompts.")
         
     images = []
-    url = os.getenv('JAX_BACKEND_URL')
+    generator = torch.manual_seed(68781279)
     payload = {'prompt': prompt, 'negative_prompt': negative, 'guidance_scale': scale}
-    images_request = requests.post(url, json = payload)
-    for image in images_request.json()["images"]:
-        image_b64 = (f"data:image/jpeg;base64,{image}")
-        images.append(image_b64)
+    image = pipe(prompt=prompt,
+                 negative_prompt=negative,
+                 guidance_scale=scale,
+                 num_inference_steps=15,
+                 generator=generator,
+                 height=512,
+                 width=768).images[0]
+
+    images.append(image)
     
     return images
     

requirements.txt

@@ -1 +1,6 @@
-python-dotenv
+python-dotenv
+diffusers
+transformers
+accelerate
+scipy
+safetensors