tskolm commited on
Commit
4719e8c
·
1 Parent(s): 09b842c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,8 +18,8 @@ def generate(tokenizer, model, text, features):
18
 
19
 
20
  def load_model():
21
- tokenizer = torch.load('./final_model/tokenizer.pt')
22
- model = torch.load('./final_model/model.pt', map_location=torch.device('cpu'))
23
  return tokenizer, model
24
 
25
 
 
18
 
19
 
20
  def load_model():
21
+ tokenizer = torch.load('./tokenizer.pt')
22
+ model = torch.load('./model.pt', map_location=torch.device('cpu'))
23
  return tokenizer, model
24
 
25