Shreekant Kalwar (Nokia) commited on
Commit
4395cc5
·
1 Parent(s): 4815889

model change

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -25,9 +25,11 @@ class ChatRequest(BaseModel):
25
  message: str
26
 
27
  # Load DeepSeek model (small one for local use)
28
- #model_name = "deepseek-ai/deepseek-coder-1.3b-instruct"
29
 
30
- model_name="Qwen/Qwen2.5-1.5B-Instruct"
 
 
31
  #model_name="TinyLlama/TinyLlama-1.1B-Chat-v1.0"
32
 
33
  print("Loading model... this may take a minute ⏳")
 
25
  message: str
26
 
27
  # Load DeepSeek model (small one for local use)
28
+ model_name = "deepseek-ai/deepseek-coder-1.3b-base"
29
 
30
+ # model_name = "deepseek-ai/deepseek-llm-7b-base"
31
+
32
+ #model_name="Qwen/Qwen2.5-1.5B-Instruct"
33
  #model_name="TinyLlama/TinyLlama-1.1B-Chat-v1.0"
34
 
35
  print("Loading model... this may take a minute ⏳")