Update app.py
Browse files
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:
|
| 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 |
|