Update app.py

app.py

@@ -8,11 +8,12 @@ openai_client = OpenAI(api_key=OPENAI_API_KEY)
 DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
 deepseek_client = OpenAI(api_key=DEEPSEEK_API_KEY, base_url="https://api.deepseek.com")
 
-def generate_response(prompt,temperature, top_p, max_tokens, repetition_penalty):
+def generate_response(prompt,model_provider,temperature, top_p, max_tokens, repetition_penalty):
     try:
         response = openai_client.chat.completions.create(
-            model="gpt-3.5-turbo",  # or another model of your choice
+            model="gpt-3.5-turbo",  # model="deepseek-chat"
             messages=[{"role": "user", "content": prompt}],
+            model_provider=model_provider
             temperature=temperature,
             top_p=top_p,
             max_tokens=max_tokens,
@@ -26,6 +27,7 @@ def generate_response(prompt,temperature, top_p, max_tokens, repetition_penalty)
 iface = gr.Interface(
     fn=generate_response,
     inputs=[
+        gr.Dropdown(choices=["DeepSeek", "OpenAI"], value="DeepSeek", label="Model Provider"),
         gr.Textbox(label="Prompt", lines=6, placeholder="Ask something..."),
         gr.Slider(minimum=0.1, maximum=1.5, value=0.7, step=0.1, label="Temperature"),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.9, step=0.05, label="Top-p"),