khan123 commited on
Commit
cdf69a2
1 Parent(s): 141f72f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers import pipeline
7
  from datasets import Dataset, Audio
8
  import gradio as gr
9
  import torch
10
- from utils.postprocess import text_mapping
11
  import json
12
  import os
13
  # Set model & Dataset NM
@@ -49,6 +49,7 @@ def gaepago_fn(tmp_audio_dir):
49
  # add postprocessing
50
  ## 1. text mapping
51
  output = text_mapping(predicted_label,text_label)
 
52
  return output
53
 
54
  # Main
 
7
  from datasets import Dataset, Audio
8
  import gradio as gr
9
  import torch
10
+ from utils.postprocess import text_mapping,text_encoding
11
  import json
12
  import os
13
  # Set model & Dataset NM
 
49
  # add postprocessing
50
  ## 1. text mapping
51
  output = text_mapping(predicted_label,text_label)
52
+ output = text_encoding(output)
53
  return output
54
 
55
  # Main