deepkansara-123 commited on
Commit
7d9290b
·
verified ·
1 Parent(s): b183b9a

Update app.py

Browse files

try to file name propre

Files changed (1) hide show
  1. app.py +2 -3
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)