Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ import uuid
|
|
10 |
from sentence_transformers import SentenceTransformer
|
11 |
import os
|
12 |
|
13 |
-
|
14 |
model_name = 'google/flan-t5-base'
|
15 |
model = T5ForConditionalGeneration.from_pretrained(model_name, device_map='auto', offload_folder="offload")
|
16 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
@@ -44,7 +44,7 @@ def local_query(query, context):
|
|
44 |
|
45 |
def run_query(history, query):
|
46 |
|
47 |
-
|
48 |
|
49 |
loader = PDFMinerLoader(file_name)
|
50 |
doc = loader.load()
|
@@ -90,7 +90,7 @@ def run_query(history, query):
|
|
90 |
def upload_pdf(file):
|
91 |
try:
|
92 |
if file is not None:
|
93 |
-
|
94 |
return 'Successfully uploaded!'
|
95 |
else:
|
96 |
return "No file uploaded."
|
|
|
10 |
from sentence_transformers import SentenceTransformer
|
11 |
import os
|
12 |
|
13 |
+
file f= ''
|
14 |
model_name = 'google/flan-t5-base'
|
15 |
model = T5ForConditionalGeneration.from_pretrained(model_name, device_map='auto', offload_folder="offload")
|
16 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
|
44 |
|
45 |
def run_query(history, query):
|
46 |
|
47 |
+
file_name = f.name
|
48 |
|
49 |
loader = PDFMinerLoader(file_name)
|
50 |
doc = loader.load()
|
|
|
90 |
def upload_pdf(file):
|
91 |
try:
|
92 |
if file is not None:
|
93 |
+
f = file
|
94 |
return 'Successfully uploaded!'
|
95 |
else:
|
96 |
return "No file uploaded."
|