barto17 commited on
Commit
ec1a913
·
1 Parent(s): 8299dc1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -2,6 +2,12 @@ import torch
2
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
3
  from transformers.pipelines.audio_utils import ffmpeg_read
4
  import gradio as gr
 
 
 
 
 
 
5
 
6
  LANGUANGE_MAP = {
7
  0: 'Arabic',
 
2
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
3
  from transformers.pipelines.audio_utils import ffmpeg_read
4
  import gradio as gr
5
+ from transformers import WhisperForConditionalGeneration, WhisperProcessor
6
+ from transformers.models.whisper.tokenization_whisper import LANGUAGES
7
+ from transformers.pipelines.audio_utils import ffmpeg_read
8
+
9
+ model_id = "openai/whisper-large-v2"
10
+
11
 
12
  LANGUANGE_MAP = {
13
  0: 'Arabic',