Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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,
|