hypeconqueror1 commited on
Commit
aad567b
·
verified ·
1 Parent(s): 1713b7b

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -17,10 +17,10 @@ async def home():
17
  @app.post('/PromptBuddy')
18
  async def PromptLLM(file: UploadFile = File(...), query: str = Form(...)):
19
 
20
- with open(file.filename, 'w') as file:
21
- file.write('pdf_file')
22
 
23
- loader = PyMuPDFLoader(file_path='pdf_file')
24
  data = loader.load()
25
 
26
  # Create embeddings using Sentence Transformers
 
17
  @app.post('/PromptBuddy')
18
  async def PromptLLM(file: UploadFile = File(...), query: str = Form(...)):
19
 
20
+ with open(f"uploads/{file.filename}", "wb") as f:
21
+ f.write(file.file.read())
22
 
23
+ loader = PyMuPDFLoader(file_path=f"uploads/{file.filename}")
24
  data = loader.load()
25
 
26
  # Create embeddings using Sentence Transformers