Spaces:
Running
Running
Commit ·
5b95f59
1
Parent(s): 906875e
fix(app): raise error if no audio selected
Browse files
src/emotion_analysis/app/on_callback.py
CHANGED
|
@@ -1,11 +1,15 @@
|
|
| 1 |
from pathlib import Path
|
| 2 |
|
|
|
|
|
|
|
| 3 |
from ..services import Classifier
|
| 4 |
|
| 5 |
|
| 6 |
def on_classify(model: Classifier):
|
| 7 |
def fn(audio: Path | str):
|
| 8 |
-
|
| 9 |
-
|
|
|
|
|
|
|
| 10 |
|
| 11 |
return fn
|
|
|
|
| 1 |
from pathlib import Path
|
| 2 |
|
| 3 |
+
from gradio import Error
|
| 4 |
+
|
| 5 |
from ..services import Classifier
|
| 6 |
|
| 7 |
|
| 8 |
def on_classify(model: Classifier):
|
| 9 |
def fn(audio: Path | str):
|
| 10 |
+
if not audio:
|
| 11 |
+
Error("No Audio selected")
|
| 12 |
+
return [None] * len(model.id2label.keys())
|
| 13 |
+
return [f"{prob:.03%}" for prob in model.predict(audio)]
|
| 14 |
|
| 15 |
return fn
|