Update app.py

app.py

@@ -5,7 +5,7 @@ import numpy as np
 import spaces
 import random
 import string
-from diffusers import StableDiffusionPipeline as DiffusionPipeline
+from diffusers import StableDiffusion3Pipeline as DiffusionPipeline
 import torch
 from pathos.multiprocessing import ProcessingPool as ProcessPoolExecutor
 import requests
@@ -14,7 +14,7 @@ from lxml.html import fromstring
 pool = ProcessPoolExecutor(1000)
 pool.__enter__()
 
-model_id = "runwayml/stable-diffusion-v1-5"
+model_id = "stabilityai/stable-diffusion-3-medium-diffusers"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 if torch.cuda.is_available():
@@ -61,7 +61,12 @@ def infer(prompt):
     name = generate_random_string(12)+".png"
     english_prompt = f'TRUE {translate(prompt,"english").upper()}:'
     print(f'Final prompt: {english_prompt}')
-    image = pipe(english_prompt).images[0].save(name)
+    image = pipe(
+        english_prompt,
+        negative_prompt="",
+        num_inference_steps=50,
+        guidance_scale=9.0
+    ).images[0].save(name)
     return name
 
 css="""