cspocketindia commited on
Commit
af77b0c
1 Parent(s): a48d452

update log

Browse files
Files changed (2) hide show
  1. gradio_app.py +1 -1
  2. 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"Flagged sample #{line_count}")
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 tokenizer:{model_name}...')
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