add prints for checking
Browse files- handler.py +2 -0
handler.py
CHANGED
@@ -323,6 +323,7 @@ class UptakeModel:
|
|
323 |
class FocusingQuestionModel:
|
324 |
def __init__(self, device, tokenizer, input_builder, max_length=128, path=FOCUSING_QUESTION_MODEL):
|
325 |
print("Loading models...")
|
|
|
326 |
self.device = device
|
327 |
self.tokenizer = tokenizer
|
328 |
self.input_builder = input_builder
|
@@ -369,6 +370,7 @@ def load_math_terms():
|
|
369 |
return math_terms, math_terms_dict
|
370 |
|
371 |
def run_math_density(transcript, uptake_speaker=None):
|
|
|
372 |
math_terms, math_terms_dict = load_math_terms()
|
373 |
sorted_terms = sorted(math_terms, key=len, reverse=True)
|
374 |
math_word_cloud = {}
|
|
|
323 |
class FocusingQuestionModel:
|
324 |
def __init__(self, device, tokenizer, input_builder, max_length=128, path=FOCUSING_QUESTION_MODEL):
|
325 |
print("Loading models...")
|
326 |
+
print("TEST IN FOCUSING QUESTION MODEL")
|
327 |
self.device = device
|
328 |
self.tokenizer = tokenizer
|
329 |
self.input_builder = input_builder
|
|
|
370 |
return math_terms, math_terms_dict
|
371 |
|
372 |
def run_math_density(transcript, uptake_speaker=None):
|
373 |
+
print("IN MATH DENSITY")
|
374 |
math_terms, math_terms_dict = load_math_terms()
|
375 |
sorted_terms = sorted(math_terms, key=len, reverse=True)
|
376 |
math_word_cloud = {}
|