Hugging Face's logo

Spaces:
jlondonobo
/
whisper-pt-demo
Runtime error

App Files Community
1
jlondonobo commited on
Commit
115c2df
•
1 Parent(s): 720bafa

🛠 update model and language

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -1,12 +1,11 @@
1
- import torch
2
-
3
  import gradio as gr
4
  import pytube as pt
5
- from transformers import pipeline
6
  from huggingface_hub import model_info
 
7
 
8
- MODEL_NAME = "openai/whisper-small" #this always needs to stay in line 8 :D sorry for the hackiness
9
- lang = "en"
10
 
11
  device = 0 if torch.cuda.is_available() else "cpu"
12
 
 
 
 
1
  import gradio as gr
2
  import pytube as pt
3
+ import torch
4
  from huggingface_hub import model_info
5
+ from transformers import pipeline
6
 
7
+ MODEL_NAME = "jlondonobo/whisper-medium-pt" #this always needs to stay in line 8 :D sorry for the hackiness
8
+ lang = "pt"
9
 
10
  device = 0 if torch.cuda.is_available() else "cpu"
11