thak123 commited on
Commit
0357372
·
verified ·
1 Parent(s): db503ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -11,10 +11,10 @@ m = SBDModelONNX.from_pretrained("sbd_multi_lang")
11
 
12
  def sentence_boundary_detection(input_texts):
13
  # Run inference
14
- results: List[List[str]] = m.infer(input_texts)
15
  print(results)
16
- sentences = "\n".join(results[0])
17
- return sentences, len(results[0])
18
 
19
 
20
  # Gradio interface
 
11
 
12
  def sentence_boundary_detection(input_texts):
13
  # Run inference
14
+ results: List[List[str]] = m.infer([input_texts])
15
  print(results)
16
+ sentences = "\n".join(results)
17
+ return sentences, len(results)
18
 
19
 
20
  # Gradio interface