Spaces:
Sleeping
Sleeping
Update app.py
Browse filestry to file name propre
app.py
CHANGED
|
@@ -7,8 +7,7 @@ import PyPDF2
|
|
| 7 |
import numpy as np
|
| 8 |
from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
|
| 9 |
from sklearn.metrics.pairwise import cosine_similarity
|
| 10 |
-
|
| 11 |
-
# Local imports
|
| 12 |
from database1 import create_db
|
| 13 |
from first1 import pdf_query
|
| 14 |
|
|
@@ -30,7 +29,7 @@ def upload_pdf(files):
|
|
| 30 |
messages = []
|
| 31 |
|
| 32 |
for file in files:
|
| 33 |
-
filename = file.name
|
| 34 |
token = str(uuid.uuid4())
|
| 35 |
|
| 36 |
pdf_reader = PyPDF2.PdfReader(file)
|
|
|
|
| 7 |
import numpy as np
|
| 8 |
from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
|
| 9 |
from sklearn.metrics.pairwise import cosine_similarity
|
| 10 |
+
import os
|
|
|
|
| 11 |
from database1 import create_db
|
| 12 |
from first1 import pdf_query
|
| 13 |
|
|
|
|
| 29 |
messages = []
|
| 30 |
|
| 31 |
for file in files:
|
| 32 |
+
filename = os.path.basename(file.name) # <-- FIXED HERE
|
| 33 |
token = str(uuid.uuid4())
|
| 34 |
|
| 35 |
pdf_reader = PyPDF2.PdfReader(file)
|