r1208 commited on
Commit
fa79f94
·
verified ·
1 Parent(s): 320473d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  import streamlit as st
2
  from transformers import pipeline
3
  from PIL import Image
@@ -27,7 +29,7 @@ import torch
27
  model_id = "r1208/c4ai-command-r-v01-4bit_32r"
28
 
29
  model = AutoPeftModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16)
30
- tokenizer = AutoTokenizer.from_pretrained(model_id)
31
 
32
  tokenizer_with_prefix_space = AutoTokenizer.from_pretrained(model_id, add_prefix_space=True)
33
 
 
1
+ import { oauthLoginUrl, oauthHandleRedirectIfPresent } from "@huggingface/hub";
2
+
3
  import streamlit as st
4
  from transformers import pipeline
5
  from PIL import Image
 
29
  model_id = "r1208/c4ai-command-r-v01-4bit_32r"
30
 
31
  model = AutoPeftModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16)
32
+ tokenizer = AutoTokenizer.from_pretrained(model_id, token = access_token)
33
 
34
  tokenizer_with_prefix_space = AutoTokenizer.from_pretrained(model_id, add_prefix_space=True)
35