Update app.py

app.py

@@ -12,17 +12,21 @@ model_cache = {}
 def zero_shot_classify(audio_path: str, class_names: str, model_name: str) -> Dict[str, float]:
     class_names = class_names.split(",")
     print("model name: ", model_name)
-    tokenizer = get_tokenizer(multilingual=".en" not in model_name)
-    print("tokenizer ", tokenizer)
-
+    
+    # Specify the path to the "whisper-small-esc50" model
+    model_path = "mskov/whisper-small-esc50"  
+    
     if model_name not in model_cache:
-        model = whisper.load_model(model_name)
+        model = whisper.load_model(model_path)
         print("whisper model not in model_cache ", model)
         model_cache[model_name] = model
     else:
         model = model_cache[model_name]
         print("model is in cache ", model)
 
+    # Rest of your code remains the same
+    tokenizer = get_tokenizer(multilingual=".en" not in model_name)
+    
     internal_lm_average_logprobs = classify.calculate_internal_lm_average_logprobs(
         model=model,
         class_names=class_names,
@@ -73,7 +77,7 @@ def main():
             gr.Audio(label="Input Audio",show_label=False,source="microphone",type="filepath"),
             gr.Textbox(lines=1, label="Candidate class names (comma-separated)"),
             gr.Radio(
-                choices=["tiny", "base", "small", "medium", "large"],
+                choices=["whisper-small-esc50"],
                 value="small",
                 label="Model Name",
             ),