Update app.py

app.py

@@ -1,5 +1,17 @@
 import gradio as gr
 from transformers import pipeline
+import torch
+from transformers import pipeline
+from datasets import load_dataset
+
+device = "cuda:0" if torch.cuda.is_available() else "cpu"
+
+pipe = pipeline(
+  "automatic-speech-recognition",
+  model="openai/whisper-small",
+  chunk_length_s=30,
+  device=device,
+)
 
 pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")