r1208 commited on
Commit
21ec5ae
·
verified ·
1 Parent(s): 26ea541

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -59,7 +59,7 @@ def main():
59
  model_id = "r1208/c4ai-command-r-v01-4bit_32r"
60
 
61
  model = AutoPeftModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, use_auth_token=hf_token, device_map = "auto")
62
- tokenizer = AutoTokenizer.from_pretrained(model_id, token = access_token, use_auth_token=hf_token)
63
 
64
  tokenizer_with_prefix_space = AutoTokenizer.from_pretrained(model_id, add_prefix_space=True, use_auth_token=hf_token)
65
 
@@ -75,7 +75,7 @@ def main():
75
 
76
 
77
  st.subheader("Enter text to translate")
78
- input_text = st.text_area("Text to Translate", value= {text_default}, height=300)
79
 
80
  if st.button("Translate"):
81
  if input_text:
 
59
  model_id = "r1208/c4ai-command-r-v01-4bit_32r"
60
 
61
  model = AutoPeftModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, use_auth_token=hf_token, device_map = "auto")
62
+ tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=hf_token)
63
 
64
  tokenizer_with_prefix_space = AutoTokenizer.from_pretrained(model_id, add_prefix_space=True, use_auth_token=hf_token)
65
 
 
75
 
76
 
77
  st.subheader("Enter text to translate")
78
+ input_text = st.text_area("Text to Translate", value= "Korean text here", height=300)
79
 
80
  if st.button("Translate"):
81
  if input_text: