Spaces:

bibekyess
/

bgpt

Sleeping

bibekyess commited on May 10, 2023

Commit

3e07b40

•

1 Parent(s): 3e64c24

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,10 +8,10 @@ from chat import generate_response
 if "tokenizer" not in st.session_state:
     st.session_state["tokenizer"] = AutoTokenizer.from_pretrained(
-        "MBZUAI/LaMini-Flan-T5-248M"
     )
     st.session_state["model"] = AutoModelForSeq2SeqLM.from_pretrained(
-        "MBZUAI/LaMini-Flan-T5-248M"
     )
 st.title("B-Bot : Bibek's Personal Chatbot")

 if "tokenizer" not in st.session_state:
     st.session_state["tokenizer"] = AutoTokenizer.from_pretrained(
+        "MBZUAI/LaMini-Flan-T5-783M"
     )
     st.session_state["model"] = AutoModelForSeq2SeqLM.from_pretrained(
+        "MBZUAI/LaMini-Flan-T5-783M"
     )
 st.title("B-Bot : Bibek's Personal Chatbot")