Update app.py
Browse files
app.py
CHANGED
|
@@ -68,8 +68,8 @@ def load_model():
|
|
| 68 |
raise
|
| 69 |
|
| 70 |
# Load the model and tokenizer locally
|
| 71 |
-
model, tokenizer = load_model()
|
| 72 |
-
|
| 73 |
|
| 74 |
|
| 75 |
class BasicAgent:
|
|
|
|
| 68 |
raise
|
| 69 |
|
| 70 |
# Load the model and tokenizer locally
|
| 71 |
+
#model, tokenizer = load_model()
|
| 72 |
+
model = HfApiModel(model_id='mistralai/Mistral-7B-Instruct-v0.2', max_tokens=512)
|
| 73 |
|
| 74 |
|
| 75 |
class BasicAgent:
|