Switched to smaller model

app.py

@@ -2,8 +2,13 @@ import logging
 
 import gradio as gr
 
-import torch
-from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline
+# import torch
+from transformers import (
+    AutoModelForSpeechSeq2Seq,
+    AutoProcessor,
+    pipeline,
+    WhisperProcessor,
+)
 
 
 device = "cpu"
@@ -21,7 +26,9 @@ model = AutoModelForSpeechSeq2Seq.from_pretrained(
 )
 model.to(device)
 
-processor = AutoProcessor.from_pretrained(model_id)
+processor = WhisperProcessor.from_pretrained("openai/whisper-base.en")
+
+# processor = AutoProcessor.from_pretrained(model_id)
 
 pipe = pipeline(
     task="automatic-speech-recognition",