Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -66,7 +66,7 @@ def local_css():
|
|
66 |
@st.cache_resource
|
67 |
def load_model():
|
68 |
# Using VietAI's Vietnamese GPT model
|
69 |
-
model_name = "
|
70 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
71 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
72 |
return model, tokenizer
|
|
|
66 |
@st.cache_resource
|
67 |
def load_model():
|
68 |
# Using VietAI's Vietnamese GPT model
|
69 |
+
model_name = "google-t5/t5-base"
|
70 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
71 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
72 |
return model, tokenizer
|