hylee719 commited on
Commit
9423570
·
1 Parent(s): 030dc40

add prints for checking

Browse files
Files changed (1) hide show
  1. 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 = {}