Spaces:
Paused
Paused
Update app.py
Browse files
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
|