dammy commited on
Commit
d9b8791
·
1 Parent(s): f6a1dd1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -10,7 +10,7 @@ import uuid
10
  from sentence_transformers import SentenceTransformer
11
  import os
12
 
13
- file_name = ''
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
- # file_name = file.name
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
- file_name = file.name
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."