Update app.py

app.py

@@ -11,15 +11,15 @@ import uuid
 from sentence_transformers import SentenceTransformer
 import os
 
-# model_name = 'google/flan-t5-base'
-# model = T5ForConditionalGeneration.from_pretrained(model_name, device_map='auto', offload_folder="offload")
-# tokenizer = AutoTokenizer.from_pretrained(model_name)
-# print('flan read')
+model_name = 'google/flan-t5-base'
+model = T5ForConditionalGeneration.from_pretrained(model_name, device_map='auto', offload_folder="offload")
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+print('flan read')
 
 
-# ST_name = 'sentence-transformers/sentence-t5-base'
-# st_model = SentenceTransformer(ST_name)
-# print('sentence read')
+ST_name = 'sentence-transformers/sentence-t5-base'
+st_model = SentenceTransformer(ST_name)
+print('sentence read')
 
 
 def get_context(query_text):