Changed the moedl to the newly trained one

app.py

@@ -17,9 +17,13 @@ from nltk.tokenize import word_tokenize
 @st.cache_resource
 def get_models(llama=False):
   st.write('Loading the model...')
-  config = PeftConfig.from_pretrained("NursNurs/T5ForReverseDictionary")
+  # config = PeftConfig.from_pretrained("NursNurs/T5ForReverseDictionary")
+  # model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
+  # model = PeftModel.from_pretrained(model, "NursNurs/T5ForReverseDictionary")
+
+  config = PeftConfig.from_pretrained("YouNameIt/T5ForReverseDictionary_prefix_tuned")
   model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
-  model = PeftModel.from_pretrained(model, "NursNurs/T5ForReverseDictionary")
+  model = PeftModel.from_pretrained(model, "YouNameIt/T5ForReverseDictionary_prefix_tuned")
 
   tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")