Spaces:
Sleeping
Sleeping
Update pages/1_Descriptive_chatbot.py
Browse files
pages/1_Descriptive_chatbot.py
CHANGED
|
@@ -18,9 +18,6 @@ from nltk.tokenize import word_tokenize
|
|
| 18 |
@st.cache_resource
|
| 19 |
def get_models(llama=False):
|
| 20 |
st.write('Loading the model...')
|
| 21 |
-
# config = PeftConfig.from_pretrained("NursNurs/T5ForReverseDictionary")
|
| 22 |
-
# model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
|
| 23 |
-
# model = PeftModel.from_pretrained(model, "NursNurs/T5ForReverseDictionary")
|
| 24 |
|
| 25 |
config = PeftConfig.from_pretrained("YouNameIt/T5ForReverseDictionary_prefix_tuned")
|
| 26 |
model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
|
|
@@ -238,7 +235,6 @@ if 'descriptions' not in st.session_state:
|
|
| 238 |
|
| 239 |
st.title("You name it! 🗣")
|
| 240 |
|
| 241 |
-
# JS: would remove Simon by some neutral avatar
|
| 242 |
with st.chat_message('user', avatar='nursulu.jpg'):
|
| 243 |
st.write("Hey assistant!")
|
| 244 |
|
|
|
|
| 18 |
@st.cache_resource
|
| 19 |
def get_models(llama=False):
|
| 20 |
st.write('Loading the model...')
|
|
|
|
|
|
|
|
|
|
| 21 |
|
| 22 |
config = PeftConfig.from_pretrained("YouNameIt/T5ForReverseDictionary_prefix_tuned")
|
| 23 |
model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
|
|
|
|
| 235 |
|
| 236 |
st.title("You name it! 🗣")
|
| 237 |
|
|
|
|
| 238 |
with st.chat_message('user', avatar='nursulu.jpg'):
|
| 239 |
st.write("Hey assistant!")
|
| 240 |
|