Pujan-Dev commited on
Commit
91c5168
·
verified ·
1 Parent(s): 4ecc57f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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