mskov commited on
Commit
98a02af
·
1 Parent(s): 63a5f5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -13,12 +13,15 @@ def zero_shot_classify(audio_path: str, class_names: str, model_name: str) -> Di
13
  class_names = class_names.split(",")
14
  print("model name: ", model_name)
15
  tokenizer = get_tokenizer(multilingual=".en" not in model_name)
 
16
 
17
  if model_name not in model_cache:
18
  model = whisper.load_model(model_name)
 
19
  model_cache[model_name] = model
20
  else:
21
  model = model_cache[model_name]
 
22
 
23
  internal_lm_average_logprobs = classify.calculate_internal_lm_average_logprobs(
24
  model=model,
 
13
  class_names = class_names.split(",")
14
  print("model name: ", model_name)
15
  tokenizer = get_tokenizer(multilingual=".en" not in model_name)
16
+ print("tokenizer ", tokenizer)
17
 
18
  if model_name not in model_cache:
19
  model = whisper.load_model(model_name)
20
+ print("whisper model not in model_cache ", model)
21
  model_cache[model_name] = model
22
  else:
23
  model = model_cache[model_name]
24
+ print("model is in cache ", model)
25
 
26
  internal_lm_average_logprobs = classify.calculate_internal_lm_average_logprobs(
27
  model=model,