Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
import os
|
2 |
-
import
|
|
|
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)
|