Update app.py

app.py

@@ -59,8 +59,12 @@ def get_pre_processed_data(size):
 
 def get_top_k_matches(np_array,k,sentences):
    indices=[]
+
    for idx in heapsort(np_array,k):
-     i=len(sentences[idx])-1                  #based on the index find the sentence- reason for storing idx but not sentence
+     try:
+        i=len(sentences[idx])-1                  #based on the index find the sentence- reason for storing idx but not sentence
+     catch:
+       print(idx)
      count=1
      number=0
      while sentences[idx][i]!='@':                     #o(8-10 digits)- o(1)
@@ -78,8 +82,7 @@ df=pd.DataFrame({"Title":dataset['train']['Title'],"Plot":dataset['train']['Over
 def getOutput(text, size):
     sentences=get_pre_processed_data(int(size))
     np_array=getSimilarity(sentences,[text])
-    s=df.iloc[get_top_k_matches(np_array,2,sentences)]['Title']
-    return f"{s}."
+    return df.iloc[get_top_k_matches(np_array,2,sentences)]
 
 iface = gr.Interface(fn=getOutput, inputs=[gr.inputs.Textbox(label="Text"), gr.inputs.Number(label="size")], outputs="text")
 iface.launch(debug=True)