NaimaAqeel commited on
Commit
14afd96
·
verified ·
1 Parent(s): 8775260

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import os
2
- import fitz
 
3
  from docx import Document
4
  from sentence_transformers import SentenceTransformer
5
  import faiss
@@ -11,6 +12,8 @@ from langchain_community.embeddings import HuggingFaceEmbeddings
11
  from fastapi import FastAPI, UploadFile, File
12
  from typing import List
13
 
 
 
14
  app = FastAPI()
15
 
16
  # Function to extract text from a PDF file
@@ -101,4 +104,4 @@ async def query(text: str):
101
 
102
  if __name__ == "__main__":
103
  import uvicorn
104
- uvicorn.run(app, host="0.0.0.0", port=8001)
 
1
  import os
2
+ from dotenv import load_dotenv
3
+ import fitz # PyMuPDF
4
  from docx import Document
5
  from sentence_transformers import SentenceTransformer
6
  import faiss
 
12
  from fastapi import FastAPI, UploadFile, File
13
  from typing import List
14
 
15
+ load_dotenv() # Load environment variables from .env
16
+
17
  app = FastAPI()
18
 
19
  # Function to extract text from a PDF file
 
104
 
105
  if __name__ == "__main__":
106
  import uvicorn
107
+ uvicorn.run(app, host="0.0.0.0", port=8001)