update interface

app.py

@@ -254,8 +254,7 @@ def create_gradio_interface():
                 model_dropdown = gr.Dropdown(
                     choices=list(MODELS.keys()),
                     value=list(MODELS.keys())[0],
-                    label="Select Model",
-                    info="Choose the model for image generation"
+                    label="Select Model"
                 )
                 
                 prompt = gr.Textbox(
@@ -298,13 +297,12 @@ def create_gradio_interface():
                 
                 reference_image = gr.Image(
                     type="pil",
-                    label="Reference Image (optional)",
-                    info="Upload an image for img2img generation"
+                    label="Reference Image (optional)"
                 )
                 
                 with gr.Row():
                     generate_btn = gr.Button("Generate", variant="primary")
-                    unload_btn = gr.Button("Unload Model", variant="secondary")
+                    unload_btn = gr.Button("Unload Model")
             
             with gr.Column(scale=2):
                 output_image = gr.Image(label="Generated Image")
@@ -316,24 +314,24 @@ def create_gradio_interface():
         def update_params(model_name):
             model_config = MODELS[model_name]["parameters"]
             return [
-                gr.Slider.update(
+                gr.update(
                     minimum=model_config["height"]["min"],
                     maximum=model_config["height"]["max"],
                     value=model_config["height"]["default"],
                     step=model_config["height"]["step"]
                 ),
-                gr.Slider.update(
+                gr.update(
                     minimum=model_config["width"]["min"],
                     maximum=model_config["width"]["max"],
                     value=model_config["width"]["default"],
                     step=model_config["width"]["step"]
                 ),
-                gr.Slider.update(
+                gr.update(
                     minimum=model_config["num_inference_steps"]["min"],
                     maximum=model_config["num_inference_steps"]["max"],
                     value=model_config["num_inference_steps"]["default"]
                 ),
-                gr.Slider.update(
+                gr.update(
                     minimum=model_config["guidance_scale"]["min"],
                     maximum=model_config["guidance_scale"]["max"],
                     value=model_config["guidance_scale"]["default"]
@@ -378,7 +376,7 @@ def create_gradio_interface():
         )
     
     return interface
-
+    
 if __name__ == "__main__":
     import uvicorn
     from threading import Thread

requirements.txt

@@ -1,18 +1,20 @@
-fastapi>=0.104.0
-uvicorn>=0.24.0
-python-multipart>=0.0.6
-gradio>=4.11.0
-torch>=2.1.0
-torchvision>=0.16.0
-diffusers>=0.24.0
-transformers>=4.36.0
-accelerate>=0.25.0
-safetensors>=0.4.0
-xformers>=0.0.22.post7  # Optional but recommended for memory efficiency
-pillow>=10.0.0
-numpy>=1.24.0
-packaging>=23.2
-pydantic>=2.5.0
-tqdm>=4.66.0
-typing-extensions>=4.8.0
-python-dotenv>=1.0.0
+fastapi==0.95.2
+uvicorn==0.22.0
+python-multipart==0.0.6
+gradio==3.32.0
+torch==2.0.1
+torchvision==0.15.2
+diffusers==0.18.2
+transformers==4.30.2
+accelerate==0.20.3
+safetensors==0.3.1
+pillow==9.5.0
+numpy==1.24.3
+packaging==23.1
+pydantic==1.10.8
+tqdm==4.65.0
+typing-extensions==4.6.3
+python-dotenv==1.0.0
+
+# Optional but recommended for memory efficiency
+xformers==0.0.20