Commit
·
6985019
1
Parent(s):
28f8813
change app
Browse files
app.py
CHANGED
@@ -49,11 +49,11 @@ download_and_extract_gdrive(
|
|
49 |
# # --------------------------- Load the LanceDB Table and Models --------------------------- #
|
50 |
|
51 |
# Connect to LanceDB
|
52 |
-
DB_PATH = "
|
53 |
TABLE_NAME_1 = "enhanced_papers_pretrained_1"
|
54 |
TABLE_NAME_2 = "enhanced_papers_pretrained_2"
|
55 |
TABLE_NAME_3 = "enhanced_papers_finetuned"
|
56 |
-
|
57 |
db = lancedb.connect(DB_PATH)
|
58 |
table1 = db.open_table(TABLE_NAME_1)
|
59 |
table2 = db.open_table(TABLE_NAME_2)
|
|
|
49 |
# # --------------------------- Load the LanceDB Table and Models --------------------------- #
|
50 |
|
51 |
# Connect to LanceDB
|
52 |
+
DB_PATH = "./lancedb_directory_main"
|
53 |
TABLE_NAME_1 = "enhanced_papers_pretrained_1"
|
54 |
TABLE_NAME_2 = "enhanced_papers_pretrained_2"
|
55 |
TABLE_NAME_3 = "enhanced_papers_finetuned"
|
56 |
+
os.makedirs(DB_PATH, exist_ok=True)
|
57 |
db = lancedb.connect(DB_PATH)
|
58 |
table1 = db.open_table(TABLE_NAME_1)
|
59 |
table2 = db.open_table(TABLE_NAME_2)
|