Ahmed107 commited on
Commit
2d053ac
·
verified ·
1 Parent(s): 57a760c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from transformers import pipeline,AutoConfig,AutoModelForAudioClassification
2
 
3
  model_id = "Ahmed107/whisper-tiny-finetuned-eos"
4
 
@@ -9,7 +9,10 @@ label2id = {"NOT_EOS": "0", "EOS": "1"}
9
  # define config
10
  config = AutoConfig.from_pretrained(model_id, label2id=label2id, id2label=id2label)
11
  model = AutoModelForAudioClassification.from_pretrained(model_id, config = config)
12
- pipe = pipeline("audio-classification", model=model)
 
 
 
13
 
14
  def classify_audio(filepath):
15
  preds = pipe(filepath)
 
1
+ from transformers import pipeline,AutoConfig,AutoModelForAudioClassification,AutoFeatureExtractor
2
 
3
  model_id = "Ahmed107/whisper-tiny-finetuned-eos"
4
 
 
9
  # define config
10
  config = AutoConfig.from_pretrained(model_id, label2id=label2id, id2label=id2label)
11
  model = AutoModelForAudioClassification.from_pretrained(model_id, config = config)
12
+ feature_extractor = AutoFeatureExtractor.from_pretrained(
13
+ model_id,
14
+ )
15
+ pipe = pipeline("audio-classification", model=model,feature_extractor=feature_extractor)
16
 
17
  def classify_audio(filepath):
18
  preds = pipe(filepath)