Update main.py

main.py

@@ -154,7 +154,7 @@ async def get_completion(payload: Payload, request: Request):
             content={"message": "Server is under maintenance. Please try again later."}
         )
     
-    model_to_use = "gpt-4o-mini" if payload.model == "gpt-4o" else (payload.model if payload.model else "gpt-4o-mini")
+    model_to_use = payload.model if payload.model else "gpt-4o-mini"
     
     # Validate model availability
     if model_to_use not in available_model_ids: