Update app.py

app.py

@@ -17,14 +17,14 @@ models = [
     "models/stabilityai/stable-diffusion-2-1-unclip-small"
 ]
 
-def choose_model(model_name):
-    gr.Interface.load(model_name).launch()
+def generate_text(model_name, input_text):
+    model = gr.Interface.load(model_name, api_key=os.getenv("HUGGINGFACE_API_KEY"))
+    return model.process(input_text)
 
-model_dropdown = gr.inputs.Dropdown(models)
-interface = gr.Interface(fn=choose_model, inputs=model_dropdown, outputs=None)
+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")
 
-api_key = os.getenv("HUGGINGFACE_API_KEY")
-if api_key:
-    interface.api_key = api_key
+interface = gr.Interface(fn=generate_text, inputs=[model_dropdown, input_text], outputs=output_text)
 
 interface.launch()