Update app.py

app.py

@@ -22,8 +22,7 @@ def interact_with_space(system_prompt, user_prompt, assistant_response, model="g
         user_prompt=user_prompt,
         assistant_response=assistant_response,
         model=model,
-        api_name="/predict",
-        hf_token=hf_token
+        api_name="/predict"
     )
     return result
 
@@ -35,7 +34,16 @@ def create_ui():
         system_prompt_input = gr.Textbox(label="System Prompt", placeholder="Enter system prompt here", lines=2)
         user_prompt_input = gr.Textbox(label="User Prompt", placeholder="Enter user prompt here", lines=2)
         assistant_response_input = gr.Textbox(label="Assistant Response", placeholder="Enter assistant response here", lines=2)
-        model_input = gr.Textbox(label="Model", value="gpt-4o")  # Default model is gpt-4o
+        model_input = gr.Dropdown(
+            choices=[
+                "gpt-4o", "gpt-4o-mini", "gpt-4o-mini-audio-preview", "gpt-4o-mini-realtime-preview",
+                "o1", "o1-preview", "o1-mini", "gpt-4o-realtime-preview", "gpt-4o-audio-preview",
+                "gpt-4-turbo-preview", "gpt-3.5-turbo", "gpt-4"
+            ],
+            label="Model",
+            value="gpt-4o",  # Default value is "gpt-4o"
+            info="Pricing: Different models have different costs for input and output tokens, per call price."
+        )
 
         # Output
         output = gr.Textbox(label="Response", interactive=False)