change model
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ class BasicAgent:
|
|
| 20 |
"""
|
| 21 |
|
| 22 |
def __init__(self):
|
| 23 |
-
model_id = "
|
| 24 |
print(f"🚀 Loading model locally: {model_id}")
|
| 25 |
self.tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 26 |
self.model = AutoModelForCausalLM.from_pretrained(
|
|
|
|
| 20 |
"""
|
| 21 |
|
| 22 |
def __init__(self):
|
| 23 |
+
model_id = "microsoft/Phi-3-mini-4k-instruct"
|
| 24 |
print(f"🚀 Loading model locally: {model_id}")
|
| 25 |
self.tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 26 |
self.model = AutoModelForCausalLM.from_pretrained(
|