Uploading whisper small tamil model demo app.py

app.py

@@ -1,7 +1,10 @@
-from transformers import pipeline
+import torch
 import gradio as gr
+from transformers import pipeline
 
-pipe = pipeline(task="transcribe", model="deepakkumar07/whisper-small-tamil")
+pipe = pipeline(task="automatic-speech-recognition", 
+                model="deepakkumar07/whisper-small-tamil",
+                device="cuda" if torch.cuda.is_available() else "cpu")
 
 def transcribe(audio):
     text = pipe(audio)["text"]