VirtualOasis commited on
Commit
432b3c4
·
verified ·
1 Parent(s): c6ea530

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,7 +4,7 @@ from smolagents import (
4
  ToolCallingAgent,
5
  InferenceClientModel,
6
  LiteLLMModel,
7
- OpenAIModel,
8
  GradioUI,
9
  MCPClient
10
  )
@@ -59,7 +59,7 @@ def initialize_model(model_provider, api_key_or_token):
59
  max_tokens=5000
60
  )
61
  elif model_provider == "DeepSeek (Direct API)":
62
- return OpenAIModel(
63
  model_id="deepseek-chat",
64
  api_key=api_key_or_token,
65
  base_url="https://api.deepseek.com"
 
4
  ToolCallingAgent,
5
  InferenceClientModel,
6
  LiteLLMModel,
7
+ OpenAIServerModel,
8
  GradioUI,
9
  MCPClient
10
  )
 
59
  max_tokens=5000
60
  )
61
  elif model_provider == "DeepSeek (Direct API)":
62
+ return OpenAIServerModel(
63
  model_id="deepseek-chat",
64
  api_key=api_key_or_token,
65
  base_url="https://api.deepseek.com"