Update app.py
Browse files
app.py
CHANGED
|
@@ -83,8 +83,8 @@ def load_model():
|
|
| 83 |
|
| 84 |
# Load the model and tokenizer locally
|
| 85 |
#model, tokenizer = load_model()
|
| 86 |
-
model = HfApiModel()#model_id=MODEL_NAME, max_tokens=512)
|
| 87 |
-
|
| 88 |
|
| 89 |
class BasicAgent:
|
| 90 |
def __init__(self):
|
|
|
|
| 83 |
|
| 84 |
# Load the model and tokenizer locally
|
| 85 |
#model, tokenizer = load_model()
|
| 86 |
+
#model = HfApiModel()#model_id=MODEL_NAME, max_tokens=512)
|
| 87 |
+
llm_engine = load_model(MODEL_NAME)
|
| 88 |
|
| 89 |
class BasicAgent:
|
| 90 |
def __init__(self):
|