NursNurs commited on
Commit
a3d3a4d
·
1 Parent(s): 44bb176

Update pages/1_Descriptive_chatbot.py

Browse files
Files changed (1) hide show
  1. pages/1_Descriptive_chatbot.py +0 -4
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