robzchhangte commited on
Commit
b731cc5
·
verified ·
1 Parent(s): f4d4037

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -7,7 +7,8 @@ token = os.getenv("hf_token")
7
 
8
  @st.cache_resource
9
  def load_model():
10
- model_name = "robzchhangte/bloomz-dv5-with-mztok"
 
11
  tokenizer = AutoTokenizer.from_pretrained("robzchhangte/bloomz-dv5-with-mztok", use_auth_token=token)
12
  model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=token)
13
  return tokenizer, model
@@ -18,7 +19,7 @@ st.title("📝 Mizo Text Generator")
18
 
19
  prompt = st.text_area("Enter your prompt (in Mizo):", height=150)
20
 
21
- st.text("Example: Mizoram hi tlangram a ni a..")
22
 
23
  generate_button = st.button("Generate Text")
24
 
 
7
 
8
  @st.cache_resource
9
  def load_model():
10
+ # model_name = "robzchhangte/bloomz-dv5-with-mztok"
11
+ model_name = "robzchhangte/ct-gemma-2b-bnb-dv5"
12
  tokenizer = AutoTokenizer.from_pretrained("robzchhangte/bloomz-dv5-with-mztok", use_auth_token=token)
13
  model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=token)
14
  return tokenizer, model
 
19
 
20
  prompt = st.text_area("Enter your prompt (in Mizo):", height=150)
21
 
22
+ st.text("Example: Lirthei pung nasa chu hmasawnna rah a nih rualin harsatna tam tak..")
23
 
24
  generate_button = st.button("Generate Text")
25