Update app.py

app.py

@@ -5,7 +5,8 @@ from typing import List, Tuple
 
 # Define available models
 AVAILABLE_MODELS = {
-    "DeepSeek V3": "deepseek-ai/DeepSeek-V3",
+    "DeepSeek V3 (Hyperbolic.xyz)": "deepseek-ai/DeepSeek-V3",
+    "DeepSeek V3 (HuggingFace.co)": "deepseek-ai/DeepSeek-V3",
     "Llama3.3-70b-Instruct": "meta-llama/Llama-3.3-70B-Instruct",
     "Llama3.1-8b-Instruct": "meta-llama/Meta-Llama-3.1-8B-Instruct",
 }
@@ -16,8 +17,6 @@ HYPERB_API_KEY = os.getenv('HYPERBOLIC_XYZ_KEY')
 HF_API_KEY = os.getenv('HF_KEY')
 PASSWORD = os.getenv("PASSWD")  # Store the password in an environment variable
 
-DEPLOY_TO_HF = ["deepseek-ai/DeepSeek-V3"]
-
 hyperb_client = OpenAI(base_url=HYPERB_ENDPOINT_URL, api_key=HYPERB_API_KEY)
 hf_client = OpenAI(base_url=HF_ENDPOINT_URL, api_key=HF_API_KEY)
 
@@ -39,8 +38,8 @@ def respond(
     messages.append({"role": "user", "content": message})
 
     response = ""
-    
-    if model_choice in DEPLOY_TO_HF:
+
+    if "(HuggingFace.co)" in model_choice:
         this_client = hf_client
     else:
         this_client = hyperb_client