magdap116 commited on
Commit
144ef97
·
verified ·
1 Parent(s): 496dd0b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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):