ruanchaves commited on
Commit
0062690
1 Parent(s): da00f10

segmentation clear

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -123,7 +123,7 @@ def predict(s1, language, use_reranker, topk, steps):
123
 
124
  output_df = output_df.drop_duplicates(subset="segmentation", keep="first")
125
 
126
- return top_segmentation, output_df
127
 
128
 
129
  inputs = [
@@ -135,7 +135,7 @@ inputs = [
135
  ]
136
 
137
  outputs = [
138
- gr.Textbox(label="Suggested segmentation"),
139
  gr.DataFrame(label="Top alternatives"),
140
  ]
141
 
 
123
 
124
  output_df = output_df.drop_duplicates(subset="segmentation", keep="first")
125
 
126
+ return "# **{0}**".format(top_segmentation), output_df
127
 
128
 
129
  inputs = [
 
135
  ]
136
 
137
  outputs = [
138
+ gr.Markdown(label="Suggested segmentation"),
139
  gr.DataFrame(label="Top alternatives"),
140
  ]
141