chalana2001 commited on
Commit
bbf26c0
·
verified ·
1 Parent(s): 081cab3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ class PromptRequest(BaseModel):
22
  MODEL_PATH = "./"
23
 
24
  # Load tokenizer and model
25
- tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH, use_fast=True)
26
  model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_PATH)
27
 
28
  @app.get("/")
 
22
  MODEL_PATH = "./"
23
 
24
  # Load tokenizer and model
25
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH, use_fast=False)
26
  model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_PATH)
27
 
28
  @app.get("/")