omniverse1 commited on
Commit
6487066
·
verified ·
1 Parent(s): b8069f0

update app

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -36,7 +36,10 @@ def load_model():
36
  device_map="auto",
37
  trust_remote_code=True
38
  )
39
- tokenizer = AutoTokenizer.from_pretrained("amazon/chronos-bolt-base")
 
 
 
40
  return model, tokenizer
41
 
42
  # Initialize model
 
36
  device_map="auto",
37
  trust_remote_code=True
38
  )
39
+ tokenizer = AutoTokenizer.from_pretrained(
40
+ "amazon/chronos-bolt-base",
41
+ trust_remote_code=True
42
+ )
43
  return model, tokenizer
44
 
45
  # Initialize model