jpdiazpardo commited on
Commit
e14dab8
1 Parent(s): d63c5d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -13,7 +13,7 @@ from functions.timestamp import format_timestamp
13
  from functions.youtube import get_youtube_video_id
14
  #---------------------------------------------------------------------
15
 
16
- MODEL_NAME = "openai/whisper-medium"
17
  BATCH_SIZE = 8
18
  device = 0 if torch.cuda.is_available() else "cpu"
19
 
@@ -27,9 +27,8 @@ pipe = pipeline(
27
 
28
  #Formating---------------------------------------------------------------------------------------------
29
  title = "Whisper Demo: Transcribe Audio"
30
- MODEL_NAME1 = "jpdiazpardo/whisper-tiny-metal"
31
  description = ("Transcribe long-form audio inputs with the click of a button! Demo uses the"
32
- f" checkpoint [{MODEL_NAME1}](https://huggingface.co/{MODEL_NAME1}) and 🤗 Transformers to transcribe audio files"
33
  " of arbitrary length. Check some of the 'cool' examples below")
34
 
35
  linkedin = generate_icon("linkedin")
 
13
  from functions.youtube import get_youtube_video_id
14
  #---------------------------------------------------------------------
15
 
16
+ MODEL_NAME = "jpdiazpardo/whisper-tiny-metal"
17
  BATCH_SIZE = 8
18
  device = 0 if torch.cuda.is_available() else "cpu"
19
 
 
27
 
28
  #Formating---------------------------------------------------------------------------------------------
29
  title = "Whisper Demo: Transcribe Audio"
 
30
  description = ("Transcribe long-form audio inputs with the click of a button! Demo uses the"
31
+ f" checkpoint [{MODEL_NAME}](https://huggingface.co/{MODEL_NAME}) and 🤗 Transformers to transcribe audio files"
32
  " of arbitrary length. Check some of the 'cool' examples below")
33
 
34
  linkedin = generate_icon("linkedin")