jeffeux commited on
Commit
0f04f5f
·
1 Parent(s): e120c09
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,9 +16,9 @@ def model_init():
16
  # device_map="auto",
17
  # Ref. for `half`: Chan-Jan, Thanks!
18
  ).eval().to(device)
19
- return tok, model
20
 
21
- tok, model = model_init()
22
 
23
  st.text(str(repr(model)))
24
 
 
16
  # device_map="auto",
17
  # Ref. for `half`: Chan-Jan, Thanks!
18
  ).eval().to(device)
19
+ return tokenizer, model
20
 
21
+ tokenizer, model = model_init()
22
 
23
  st.text(str(repr(model)))
24