mohamed1ai commited on
Commit
e365069
1 Parent(s): efb518e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,11 +8,11 @@ def transcrire_audio(audio, prompt):
8
  input_features = processor(audio, return_tensors="pt").input_features
9
 
10
  output_without_prompt = model.generate(input_features)
11
- transcription_sans_prompt = processor.decode(output_without_prompt[0])
12
 
13
  prompt_ids = processor.get_prompt_ids(prompt)
14
  output_with_prompt = model.generate(input_features, prompt_ids=prompt_ids)
15
- transcription_avec_prompt = processor.decode(output_with_prompt[0])
16
 
17
  return {
18
  "Transcription sans prompt": transcription_sans_prompt,
 
8
  input_features = processor(audio, return_tensors="pt").input_features
9
 
10
  output_without_prompt = model.generate(input_features)
11
+ transcription_sans_prompt = processor.decode(output_without_prompt[0], skip_special_tokens=True)
12
 
13
  prompt_ids = processor.get_prompt_ids(prompt)
14
  output_with_prompt = model.generate(input_features, prompt_ids=prompt_ids)
15
+ transcription_avec_prompt = processor.decode(output_with_prompt[0], skip_special_tokens=True)
16
 
17
  return {
18
  "Transcription sans prompt": transcription_sans_prompt,