Update app.py
Browse files
app.py
CHANGED
|
@@ -74,7 +74,7 @@ def load_model():
|
|
| 74 |
|
| 75 |
# Load the model and tokenizer locally
|
| 76 |
#model, tokenizer = load_model()
|
| 77 |
-
model = HfApiModel(model_id=
|
| 78 |
|
| 79 |
|
| 80 |
class BasicAgent:
|
|
|
|
| 74 |
|
| 75 |
# Load the model and tokenizer locally
|
| 76 |
#model, tokenizer = load_model()
|
| 77 |
+
model = HfApiModel(model_id=MODEL_NAME, max_tokens=512)
|
| 78 |
|
| 79 |
|
| 80 |
class BasicAgent:
|