Update app.py

app.py

@@ -1,11 +1,7 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-def update_model(selected_model, use_textbox, custom_model):
-    if use_textbox:
-        return custom_model  # Return the custom model from textbox
-    else:
-        return selected_model  # Return the selected model from dropdown
+
 
 # Initialize Hugging Face Inference Client
 def get_client(model_name):
@@ -19,66 +15,26 @@ def generate_image(prompt, model_name):
 
 # Gradio interface
 with gr.Blocks() as demo:
-    # List of available models (custom models included)
-    model_options = {
-        "Stable Diffusion 2": "stabilityai/stable-diffusion-2",
-        "Stable Diffusion 1.5": "runwayml/stable-diffusion-v1-5",
-        "DALL-E Mini": "dalle-mini/dalle-mini",
-        "FLUX 1.0 (black-forest-labs)": "black-forest-labs/FLUX.1-dev",
-        "Pony Diffusion V6": "John6666/pony-diffusion-v6-xl-sdxl-spo"
-    }
-
     gr.Markdown("# Text to Image Generator using Hugging Face Inference Client")
     
     with gr.Row():
-        use_textbox = gr.Checkbox(label="Use Custom Model", value=False)
-        
-    with gr.Row():
-        # Input for text prompt
-        prompt_input = gr.Textbox(label="Enter your prompt", placeholder="Describe the image you want...")
-        
-        # Button to generate image
-        generate_button = gr.Button("Generate Image")
-        
         with gr.Column():
             # Dropdown for model selection
-            model_dropdown = gr.Dropdown(
-                label="Select Model",
-                choices=list(model_options.keys()),  # Display model names
-                value="FLUX 1.0 (black-forest-labs)",  # Default model
-            )
-        
-        custom_model_textbox = gr.Textbox(label="Enter Custom Model", placeholder="Type your model name here", visible=False)
-
-        output = gr.Textbox(label="enter Model")
-
-        with gr.Row():
-            submit_button = gr.Button("Submit Option")
+            prompt_model = gr.Textbox(label="Enter your prompt", placeholder="your model...")
+            
+            
+            # Input for text prompt
+            prompt_input = gr.Textbox(label="Enter your prompt", placeholder="Describe the image you want...")
+            
+            # Button to generate image
+            generate_button = gr.Button("Generate Image")
         
         with gr.Column():
             # Image output
             image_output = gr.Image(label="Generated Image")
-
-    # Event to update the selected model based on the checkbox
-    use_textbox.change(
-        lambda checked: (not checked, None, "" if checked else None),
-        inputs=[use_textbox],
-        outputs=[model_dropdown, custom_model_textbox]
-    )
-    
-    # Button to confirm selection
-    submit_button.click(
-        update_model,
-        inputs=[model_dropdown, use_textbox, custom_model_textbox],
-        outputs=[output]
-    )
     
-    # Link the button click to the image generation function
-    generate_button.click(
-        generate_image,
-        inputs=[prompt_input, output],
-        outputs=image_output
-    )
+    # Link the button click to the function
+    generate_button.click(generate_image, inputs=[prompt_model, prompt_input], outputs=image_output)
 
 # Launch the Gradio app
-demo.launch(share=True)
+demo.launch()