Update app.py

app.py

@@ -10,15 +10,15 @@ import gradio as gr
 import uuid
 from sentence_transformers import SentenceTransformer
 
-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):
@@ -46,13 +46,8 @@ def run_query(query):
     return result
 
 
-    
-def upload_pdf(file):
-    # Save the uploaded file
-    file_name = file.name
-    pdf_filename = os.path.basename(file_path)
+def load_document(pdf_filename):
 
-    # Load a document
     loader = PDFMinerLoader(pdf_filename)
     doc = loader.load()
 
@@ -75,7 +70,18 @@ def upload_pdf(file):
         ids=ids
     )
 
-    return 'hello'
+    return 'Success'
+
+    
+def upload_pdf(file):
+    # Save the uploaded file
+    file_name = file.name
+    pdf_filename = os.path.basename(file_path)
+
+    messsage = load_document(pdf_filename)
+    
+
+    return messsage