Upload app.py

app.py

@@ -177,9 +177,9 @@ def get_model_answer(model_choice: str, final_prompt: str, question: str) -> str
     content_to_send = final_prompt + "\n\n" + question
 
     if model_choice == "ChatGPT 4o":
-        return call_openai_api(content_to_send, model_name="gpt-4")
+        return call_openai_api(content_to_send, model_name="gpt-4o")
     elif model_choice == "ChatGPT o1":
-        return call_openai_api(content_to_send, model_name="gpt-3.5-turbo")
+        return call_openai_api(content_to_send, model_name="o1")
     elif model_choice == "Claude 3.5 Sonnet":
         return call_claude_api(content_to_send)
     else: