Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,8 +7,7 @@ import torch
|
|
| 7 |
MODELS = {
|
| 8 |
"Qwen2.5-0.5B": "Qwen/Qwen2.5-0.5B-Instruct",
|
| 9 |
"Qwen2.5-1.5B": "Qwen/Qwen2.5-1.5B-Instruct",
|
| 10 |
-
"Phi-3-mini": "microsoft/Phi-3-mini-4k-instruct"
|
| 11 |
-
"Is It Job": "MrAlexGov/is-it-job-ru-q1a8e-lkjvk"
|
| 12 |
}
|
| 13 |
|
| 14 |
def load_model(model_key: str):
|
|
|
|
| 7 |
MODELS = {
|
| 8 |
"Qwen2.5-0.5B": "Qwen/Qwen2.5-0.5B-Instruct",
|
| 9 |
"Qwen2.5-1.5B": "Qwen/Qwen2.5-1.5B-Instruct",
|
| 10 |
+
"Phi-3-mini": "microsoft/Phi-3-mini-4k-instruct"
|
|
|
|
| 11 |
}
|
| 12 |
|
| 13 |
def load_model(model_key: str):
|