Create app.py

app.py

@@ -58,6 +58,17 @@ button:hover {
 }
 """
 
+# Define the generate_response function
+def generate_response(input_text, no_words, user_type, model_type):
+    if model_type == "DialoGPT":
+        return Get_DialoGPT_Response(input_text, no_words, user_type)
+    elif model_type == "DistilGPT":
+        return Get_DistilGPT_Response(input_text, no_words, user_type)
+    elif model_type == "MedGPT":
+        return Get_MedGPT_Response(input_text, no_words, user_type)
+    else:
+        return "Invalid model type selected."
+
 # Build Gradio interface
 with gr.Blocks(css=custom_css) as interface:
     gr.Markdown("<h1>Healthwise AI Assistant 🚑</h1>")
@@ -83,4 +94,4 @@ with gr.Blocks(css=custom_css) as interface:
 
     submit_button.click(generate_response, inputs=[input_text, no_words, user_type, model_type], outputs=response)
 
-interface.launch()
+interface.launch()