Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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,
|
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=
|
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:
|