Commit
·
929330a
1
Parent(s):
df3e017
feat: Added application file
Browse files- config/config.ini +2 -2
- src/data_loader.py +1 -1
config/config.ini
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
[data]
|
| 2 |
-
data_dir =
|
| 3 |
-
possible_labels_dir
|
| 4 |
|
| 5 |
[model]
|
| 6 |
name = facebook/bart-large-mnli
|
|
|
|
| 1 |
[data]
|
| 2 |
+
data_dir = ../data/data.json
|
| 3 |
+
possible_labels_dir =../data/possible_labels.json
|
| 4 |
|
| 5 |
[model]
|
| 6 |
name = facebook/bart-large-mnli
|
src/data_loader.py
CHANGED
|
@@ -3,7 +3,7 @@ import json
|
|
| 3 |
def get_config():
|
| 4 |
# Read config
|
| 5 |
config = configparser.ConfigParser()
|
| 6 |
-
config.read(
|
| 7 |
|
| 8 |
# Get model_name
|
| 9 |
model_name = config["model"]["name"]
|
|
|
|
| 3 |
def get_config():
|
| 4 |
# Read config
|
| 5 |
config = configparser.ConfigParser()
|
| 6 |
+
config.read("../config/config.ini")
|
| 7 |
|
| 8 |
# Get model_name
|
| 9 |
model_name = config["model"]["name"]
|