chuuhtetnaing commited on
Commit
24c6cb7
1 Parent(s): 2a688d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,5 +1,4 @@
1
  import spaces
2
- import torch
3
 
4
  import gradio as gr
5
  import yt_dlp as youtube_dl
@@ -14,7 +13,7 @@ BATCH_SIZE = 8
14
  FILE_LIMIT_MB = 1000
15
  YT_LENGTH_LIMIT_S = 3600 # limit to 1 hour YouTube files
16
 
17
- device = 0 if torch.cuda.is_available() else "cpu"
18
 
19
  pipe = pipeline(
20
  task="automatic-speech-recognition",
 
1
  import spaces
 
2
 
3
  import gradio as gr
4
  import yt_dlp as youtube_dl
 
13
  FILE_LIMIT_MB = 1000
14
  YT_LENGTH_LIMIT_S = 3600 # limit to 1 hour YouTube files
15
 
16
+ device = 0
17
 
18
  pipe = pipeline(
19
  task="automatic-speech-recognition",