Update app.py

app.py

@@ -1,4 +1,3 @@
-from transformers import pipeline
 import gradio as gr
 import openai
 
@@ -16,24 +15,24 @@ def get_openai_response(api_key, model, prompt):
 
 def generate_responses(api_key, prompt):
     response_gpt3 = get_openai_response(api_key, "gpt-3.5-turbo", prompt)
-    response_gpt4 = get_openai_response(api_key, "gpt-4o", prompt)
+    response_gpt4 = get_openai_response(api_key, "gpt-4", prompt)
     return response_gpt3, response_gpt4
 
 # Gradio Interface
 inputs = [
-    gr.inputs.Textbox(label="OpenAI API Key"),
-    gr.inputs.Textbox(label="Input")
+    gr.Textbox(label="OpenAI API Key"),
+    gr.Textbox(label="Your Question")
 ]
 
 outputs = [
-    gr.outputs.Textbox(label="GPT-3.5-Turbo Output"),
-    gr.outputs.Textbox(label="GPT-4o Output")
+    gr.Textbox(label="GPT-3.5 Turbo Response"),
+    gr.Textbox(label="GPT-4 Response")
 ]
 
 gr.Interface(
     fn=generate_responses,
     inputs=inputs,
     outputs=outputs,
-    title="Compare GPT Models Challenge",
-    description="Enter your OpenAI API Key and your question to get responses from GPT-3.5-Turbo and GPT-4o, compare their results."
+    title="AI Prompt Challenge",
+    description="Enter your OpenAI API Key and your question to get responses from GPT-3.5 Turbo and GPT-4 models."
 ).launch()