Update app.py

app.py

@@ -14,6 +14,7 @@ translator = Translator()
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
 # Constants
 model = "stabilityai/stable-diffusion-3-medium"
+repo= "stabilityai/stable-diffusion-3-medium-diffusers"
 MAX_SEED = np.iinfo(np.int32).max
 
 CSS = """
@@ -37,7 +38,7 @@ JS = """function () {
 
 # Ensure model and scheduler are initialized in GPU-enabled function
 if torch.cuda.is_available():
-    pipe = StableDiffusion3Pipeline.from_pretrained(model, torch_dtype=torch.float16, revision="refs/pr/26").to("cuda")
+    pipe = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=torch.float16).to("cuda")
 
 
 # Function 
@@ -48,7 +49,7 @@ def generate_image(
     width=1024,
     height=1024,
     scale=1.5,
-    steps=20,
+    steps=28,
     clip=3):
 
     
@@ -124,7 +125,7 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
                 minimum=1,
                 maximum=50,
                 step=1,
-                value=20,
+                value=28,
             )
             clip = gr.Slider(
                 label="Clip Skip",