mamkkl commited on
Commit
945927e
·
verified ·
1 Parent(s): d9101f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -85,7 +85,7 @@ def loadModel():
85
  # )
86
  tokenizer = AutoTokenizer.from_pretrained(base_model,use_fast=False,cache_dir=cache_dir)
87
  tokenizer.pad_token = tokenizer.unk_token
88
- return model
89
 
90
  model, tokenizer = loadModel()
91
 
 
85
  # )
86
  tokenizer = AutoTokenizer.from_pretrained(base_model,use_fast=False,cache_dir=cache_dir)
87
  tokenizer.pad_token = tokenizer.unk_token
88
+ return model, tokenizer
89
 
90
  model, tokenizer = loadModel()
91