Spaces:

Ngoufack
/

verbalens

Sleeping

Ngoufack commited on Mar 4

Commit

01daa7a

1 Parent(s): 50e3e8a

wefsdrz

Files changed (1) hide show

app.py CHANGED Viewed

@@ -34,7 +34,7 @@ def getpreferredencoding(do_setlocale = True):
 locale.getpreferredencoding = getpreferredencoding
 embedding_model = PretrainedSpeakerEmbedding(
     "speechbrain/spkrec-ecapa-voxceleb",
-    device=torch.device("cpu"))
 model = whisper.load_model(model_size)
 audio = Audio()
 def segment_embedding(segment,duration,path):
@@ -149,7 +149,7 @@ yt_transcribe = gr.Interface(
         gr.Radio(["transcribe", "translate"], label="Task", value="transcribe"),
     ],
     outputs=["html", "text"],
-    title="VerbaLend Demo with WhisperX",
     description="Transcribe YouTube videos using WhisperX.",
     allow_flagging="never",
 )

 locale.getpreferredencoding = getpreferredencoding
 embedding_model = PretrainedSpeakerEmbedding(
     "speechbrain/spkrec-ecapa-voxceleb",
+    device=torch.device("cuda"))
 model = whisper.load_model(model_size)
 audio = Audio()
 def segment_embedding(segment,duration,path):
         gr.Radio(["transcribe", "translate"], label="Task", value="transcribe"),
     ],
     outputs=["html", "text"],
+    title="VerbaLens Demo 1 : Prototyping",
     description="Transcribe YouTube videos using WhisperX.",
     allow_flagging="never",
 )