Fix remove

InvoiceableAI · Jan 23, 2023 · 2d77945 · 2d77945
1 parent 55148d5
commit 2d77945
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/main.py b/main.py
@@ -86,6 +86,7 @@ def upload():
                 file.save(os.path.join('uploads', filename))
             answers = nlp('uploads/' + filename, question)
             if len(answers) == 0:
+                os.remove(os.path.join(os.getcwd(), 'uploads/' + filename))
                 return 'Sorry, I can\'t find the answer for that question. Please try again with a different document.'
             answer = answers[0]
             os.remove(os.path.join(os.getcwd(), 'uploads/' + filename))