Rakesh30 commited on
Commit
ba3e6d5
·
1 Parent(s): 493d46f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -84,8 +84,8 @@ def getOutput(text, size=1000):
84
  np_array=getSimilarity(sentences,[text])
85
 
86
  output=[]
87
- for movie in df.iloc[get_top_k_matches(np_array,5,sentences)]:
88
- output.append(movie['Title']+'\n'+movie['Plot'])
89
  return output
90
  iface = gr.Interface(fn=getOutput,
91
  inputs=[gr.inputs.Textbox(label="Text")],
 
84
  np_array=getSimilarity(sentences,[text])
85
 
86
  output=[]
87
+ for idx in get_top_k_matches(np_array,5,sentences):
88
+ output.append(df.iloc[idx]['Title']+'\n'+df.iloc[idx]['Plot'])
89
  return output
90
  iface = gr.Interface(fn=getOutput,
91
  inputs=[gr.inputs.Textbox(label="Text")],