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):
|