Spaces:

pktpaulie
/

whisper_gradio

Sleeping

pktpaulie commited on Oct 1, 2024

Commit

19638cd

verified ·

1 Parent(s): 8445fe7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,5 +1,5 @@
 from transformers import pipeline
-import torch
 import gradio as gr
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
@@ -8,7 +8,7 @@ pipe = pipeline(
     "automatic-speech-recognition",
     model="openai/whisper-small.en",
     chunk_length_s=30,
-    device=device,
 )
 # Function to transcribe audio

 from transformers import pipeline
+# import torch
 import gradio as gr
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
     "automatic-speech-recognition",
     model="openai/whisper-small.en",
     chunk_length_s=30,
+    # device=device,
 )
 # Function to transcribe audio