Hunzla commited on
Commit
79e0339
1 Parent(s): 7987948

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -27,16 +27,15 @@ model = AutoModelForCausalLM.from_pretrained(
27
  revision="main", # Or the desired revision
28
  token=api_token
29
  )
 
30
 
31
  tokenizer = AutoTokenizer.from_pretrained(
32
  model_name,
33
  revision="main",
34
- use_auth_token=api_token
35
  # Or the desired revision
36
  )
37
 
38
- print("loaded model")
39
- tokenizer = AutoTokenizer.from_pretrained(model_name)
40
  print("loaded tokenizer")
41
  chat_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
42
  print("built pipeline")
 
27
  revision="main", # Or the desired revision
28
  token=api_token
29
  )
30
+ print("loaded model")
31
 
32
  tokenizer = AutoTokenizer.from_pretrained(
33
  model_name,
34
  revision="main",
35
+ token=api_token
36
  # Or the desired revision
37
  )
38
 
 
 
39
  print("loaded tokenizer")
40
  chat_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
41
  print("built pipeline")