Shreekant Kalwar (Nokia)
commited on
Commit
·
4395cc5
1
Parent(s):
4815889
model change
Browse files
app.py
CHANGED
|
@@ -25,9 +25,11 @@ class ChatRequest(BaseModel):
|
|
| 25 |
message: str
|
| 26 |
|
| 27 |
# Load DeepSeek model (small one for local use)
|
| 28 |
-
|
| 29 |
|
| 30 |
-
model_name="
|
|
|
|
|
|
|
| 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 ⏳")
|