Update app.py
Browse files
app.py
CHANGED
|
@@ -15,12 +15,13 @@ import fitz # PyMuPDF
|
|
| 15 |
from dotenv import load_dotenv
|
| 16 |
load_dotenv()
|
| 17 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
SECRET_TOKEN = os.getenv("SECRET_TOKEN")
|
| 19 |
bearer_scheme = HTTPBearer()
|
| 20 |
|
| 21 |
-
model_path = "./Ai-Text-Detector/model"
|
| 22 |
-
weights_path = "./Ai-Text-Detector/model_weights.pth"
|
| 23 |
-
|
| 24 |
# FastAPI app instance
|
| 25 |
app = FastAPI()
|
| 26 |
|
|
|
|
| 15 |
from dotenv import load_dotenv
|
| 16 |
load_dotenv()
|
| 17 |
|
| 18 |
+
# Define the paths to the model and weights
|
| 19 |
+
MODEL_PATH = os.getenv("MODEL_PATH", "./AI-MODEL/model")
|
| 20 |
+
WEIGHTS_PATH = os.getenv("WEIGHTS_PATH", "./AI-MODEL/model_weights.pth")
|
| 21 |
+
|
| 22 |
SECRET_TOKEN = os.getenv("SECRET_TOKEN")
|
| 23 |
bearer_scheme = HTTPBearer()
|
| 24 |
|
|
|
|
|
|
|
|
|
|
| 25 |
# FastAPI app instance
|
| 26 |
app = FastAPI()
|
| 27 |
|