lorenzoscottb commited on
Commit
5482eed
β€’
1 Parent(s): f96d1b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -82,7 +82,8 @@ pipe_S = pipeline(
82
  truncation="do_not_truncate",
83
  )
84
  def predict(text):
85
- t = pipe_S(text)
 
86
  return t
87
  interface_model_S = gr.Interface(
88
  fn=predict,
 
82
  truncation="do_not_truncate",
83
  )
84
  def predict(text):
85
+ t = pipe(text)
86
+ t = {list(dct.values())[0] : list(dct.values())[1] for dct in t[0]}
87
  return t
88
  interface_model_S = gr.Interface(
89
  fn=predict,