Update app.py

app.py

@@ -8,12 +8,11 @@ from PIL import Image
 from deep_translator import GoogleTranslator
 import json
 
-# Project by TotoB12
 
-API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
+API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-3.5-large"
 API_TOKEN = os.getenv("HF_READ_TOKEN")
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
-timeout = 100
+timeout = 1000
 
 # Function to query the API and return the generated image
 def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
@@ -71,7 +70,7 @@ css = """
 # Build the Gradio UI with Blocks
 with gr.Blocks(css=css) as app:
     # Add a title to the app
-    gr.HTML("<center><h1>FLUX.1-Schnell</h1></center>")
+    gr.HTML("<center><h1>SDXL 3.5</h1></center>")
     
     # Container for all the UI elements
     with gr.Column(elem_id="app-container"):
@@ -86,8 +85,8 @@ with gr.Blocks(css=css) as app:
                     with gr.Accordion("Advanced Settings", open=False):
                         negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="What should not be in the image", value="(deformed, distorted, disfigured), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, misspellings, typos", lines=3, elem_id="negative-prompt-text-input")
                         with gr.Row():
-                            width = gr.Slider(label="Width", value=1024, minimum=64, maximum=4096, step=32)
-                            height = gr.Slider(label="Height", value=1024, minimum=64, maximum=4096, step=32)
+                            width = gr.Slider(label="Width", value=1024, minimum=64, maximum=2048, step=32)
+                            height = gr.Slider(label="Height", value=1024, minimum=64, maximum=2048, step=32)
                         steps = gr.Slider(label="Sampling steps", value=4, minimum=1, maximum=100, step=1)
                         cfg = gr.Slider(label="CFG Scale", value=7, minimum=1, maximum=20, step=1)
                         strength = gr.Slider(label="Strength", value=0.7, minimum=0, maximum=1, step=0.001)