lucasvmigotto commited on
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
- probs: dict[int, float] = model.predict(audio, return_probs=True) # type: ignore
9
- return list(probs.values())
 
 
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