JustKiddo commited on
Commit
7fcc69d
1 Parent(s): 0bb37e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "tamgrnguyen/Gemma-2-2b-it-Vietnamese-Aesthetic"
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