Spaces:
Runtime error
Runtime error
cspocketindia
commited on
Commit
•
af77b0c
1
Parent(s):
a48d452
update log
Browse files- gradio_app.py +1 -1
- src/models/bert.py +1 -1
gradio_app.py
CHANGED
@@ -40,7 +40,7 @@ def log_record(input, output):
|
|
40 |
with open(log_file, "r", encoding="utf-8") as csvfile:
|
41 |
line_count = len([None for _ in csv.reader(csvfile)]) - 1
|
42 |
|
43 |
-
repo.push_to_hub(commit_message=f"
|
44 |
|
45 |
|
46 |
def predict(sentence):
|
|
|
40 |
with open(log_file, "r", encoding="utf-8") as csvfile:
|
41 |
line_count = len([None for _ in csv.reader(csvfile)]) - 1
|
42 |
|
43 |
+
repo.push_to_hub(commit_message=f"Logged sample #{line_count}")
|
44 |
|
45 |
|
46 |
def predict(sentence):
|
src/models/bert.py
CHANGED
@@ -16,7 +16,7 @@ class BERTClassifier():
|
|
16 |
|
17 |
|
18 |
def __init__(self, model_name="bert-base-uncased", tokenizer_name="bert-base-uncased") -> None:
|
19 |
-
print(f'Loading BERT
|
20 |
|
21 |
self.model_name = model_name
|
22 |
|
|
|
16 |
|
17 |
|
18 |
def __init__(self, model_name="bert-base-uncased", tokenizer_name="bert-base-uncased") -> None:
|
19 |
+
print(f'Loading BERT:{model_name}...')
|
20 |
|
21 |
self.model_name = model_name
|
22 |
|