Update app.py

app.py

@@ -21,23 +21,20 @@ model_list = [
 
 default_model = "stabilityai/stable-diffusion-2"
 model_name = gr.inputs.Dropdown(choices=model_list, label="Select Model", default=default_model)
-model = None
 
-def load_model(model_name):
-    global model
+def generate_image(text, model_name):
     model = gr.Interface.load(model_name, source="huggingface", api_key=api_key)
+    return model.predict(text)
 
-def predict(inputs):
-    return model.predict(inputs)
+input_text = gr.inputs.Textbox(label="Input Text")
+output_image = gr.outputs.Image(label="Generated Image")
 
 iface = gr.Interface(
-    fn=predict,
-    inputs=model_name,
-    outputs="text",
-    title="Model Selection App",
-    description="Choose a model and input data to make predictions."
+    fn=generate_image,
+    inputs=[input_text, model_name],
+    outputs=output_image,
+    title="Text to Image Generation",
+    description="Generate an image from input text using a Hugging Face model."
 )
 
-load_model(default_model)  # Load the default model initially
-
 iface.launch()