Update app.py

app.py

@@ -17,21 +17,22 @@ models = [
     "models/stabilityai/stable-diffusion-2-1-unclip-small"
 ]
 
-def generate_text(model_name, input_text):
+def generate_image(model_name, input_text):
     model = gr.Interface.load(model_name, api_key=os.getenv("HUGGINGFACE_API_KEY"))
-    return model.process(input_text)
+    output_image = model.process(input_text)
+    return output_image
 
 model_dropdown = gr.inputs.Dropdown(models, label="Select Model")
 input_text = gr.inputs.Textbox(lines=5, label="Input Text")
-output_text = gr.outputs.Textbox(label="Output Text")
+output_image = gr.outputs.Image(label="Output Image")
 
 def change_model(model_name):
     global interface
     interface.close()
-    interface = gr.Interface(generate_text, inputs=[model_name, input_text], outputs=output_text)
+    interface = gr.Interface(generate_image, inputs=[model_name, input_text], outputs=output_image)
     interface.launch()
 
-interface = gr.Interface(generate_text, inputs=[model_dropdown, input_text], outputs=output_text, capture_session=True)
+interface = gr.Interface(generate_image, inputs=[model_dropdown, input_text], outputs=output_image, capture_session=True)
 
 change_model_button = gr.Interface.load("models/stabilityai/stable-diffusion-2", capture_session=True).launch()
 interface.launch()