d

app.py

@@ -12,8 +12,8 @@ from gradio import mount_gradio_app
 
 # ------------------- Config ------------------- #
 DATA_PATH = "./dataset/pentagon_core.json"
-EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
-QA_MODEL = "facebook/bart-large-cnn"
+EMBEDDING_MODEL = "./models/all-MiniLM-L6-v2"
+QA_MODEL = "./models/bart-large-cnn"
 DEVICE = "cuda" if os.environ.get("USE_CUDA") == "1" else "cpu"
 
 # ------------------- Load Models ------------------- #