victor HF Staff commited on
Commit
2357f67
Β·
1 Parent(s): 5ab6be9

feat: Update model configuration for DeepSeek-V3 with new base URL and model name

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -16,8 +16,9 @@ import shutil
16
  # Supported models configuration
17
  MODELS = {
18
  "deepseek-ai/DeepSeek-V3": {
19
- "base_url": "https://api.deepseek.com/v1",
20
- "env_key": "DEEPSEEK_API_KEY",
 
21
  },
22
  }
23
 
@@ -243,7 +244,7 @@ Remember: Simpler is better. Only use advanced ffmpeg features if absolutely nec
243
  model_config = MODELS[model_choice]
244
  client.base_url = model_config["base_url"]
245
  client.api_key = os.environ[model_config["env_key"]]
246
- model = "deepseek-chat" if "deepseek" in model_choice.lower() else model_choice
247
 
248
  completion = client.chat.completions.create(
249
  model=model,
@@ -491,7 +492,7 @@ with gr.Blocks() as demo:
491
  gr.Markdown(
492
  """
493
  # 🏞 AI Video Composer
494
- Compose new videos from your assets using natural language. Add video, image and audio assets and let [DeepSeek-V3](https://huggingface.co/deepseek-ai/DeepSeek-V3-Base) generate a new video for you (using FFMPEG).
495
  """,
496
  elem_id="header",
497
  )
 
16
  # Supported models configuration
17
  MODELS = {
18
  "deepseek-ai/DeepSeek-V3": {
19
+ "base_url": "https://router.huggingface.co/sambanova/v1",
20
+ "env_key": "HF_TOKEN",
21
+ "model_name": "DeepSeek-V3-0324",
22
  },
23
  }
24
 
 
244
  model_config = MODELS[model_choice]
245
  client.base_url = model_config["base_url"]
246
  client.api_key = os.environ[model_config["env_key"]]
247
+ model = model_config.get("model_name", model_choice)
248
 
249
  completion = client.chat.completions.create(
250
  model=model,
 
492
  gr.Markdown(
493
  """
494
  # 🏞 AI Video Composer
495
+ Compose new videos from your assets using natural language. Add video, image and audio assets and let [DeepSeek-V3](https://huggingface.co/deepseek-ai/DeepSeek-V3-0324) generate a new video for you (using FFMPEG).
496
  """,
497
  elem_id="header",
498
  )