ddemszky commited on
Commit
25ca7c8
1 Parent(s): 3a7acca

addedd logging

Browse files
Files changed (2) hide show
  1. handler.py +6 -1
  2. test.py +2 -2
handler.py CHANGED
@@ -5,9 +5,14 @@ import weakref
5
 
6
  from utils import clean_str, clean_str_nopunct
7
  import torch
8
- from transformers import BertTokenizer, BertForSequenceClassification
9
  from utils import MultiHeadModel, BertInputBuilder, get_num_words
10
 
 
 
 
 
 
 
11
  UPTAKE_MODEL='ddemszky/uptake-model'
12
  REASONING_MODEL ='ddemszky/student-reasoning'
13
  QUESTION_MODEL ='ddemszky/question-detection'
 
5
 
6
  from utils import clean_str, clean_str_nopunct
7
  import torch
 
8
  from utils import MultiHeadModel, BertInputBuilder, get_num_words
9
 
10
+ import transformers
11
+ from transformers import BertTokenizer, BertForSequenceClassification
12
+
13
+
14
+ transformers.logging.set_verbosity_debug()
15
+
16
  UPTAKE_MODEL='ddemszky/uptake-model'
17
  REASONING_MODEL ='ddemszky/student-reasoning'
18
  QUESTION_MODEL ='ddemszky/question-detection'
test.py CHANGED
@@ -9,12 +9,12 @@ example = {
9
  "inputs": [
10
  {"uid": "1", "speaker": "Alice", "text": "How much is the fish?" },
11
  {"uid": "2", "speaker": "Bob", "text": "I do not know about the fish. Because you put a long side and it’s a long side. What do you think." },
12
- {"uid": "3", "speaker": "Alice", "text": "OK, thank you Bob." },
13
  ],
14
  "parameters": {
15
  "uptake_min_num_words": 5,
16
  "uptake_speaker": "Bob",
17
- "filename": "sample.csv",
18
  }
19
  }
20
 
 
9
  "inputs": [
10
  {"uid": "1", "speaker": "Alice", "text": "How much is the fish?" },
11
  {"uid": "2", "speaker": "Bob", "text": "I do not know about the fish. Because you put a long side and it’s a long side. What do you think." },
12
+ {"uid": "3", "speaker": "Alice", "text": "OK, thank you Bob." }
13
  ],
14
  "parameters": {
15
  "uptake_min_num_words": 5,
16
  "uptake_speaker": "Bob",
17
+ "filename": "sample.csv"
18
  }
19
  }
20