Update resultados_consulta.py

resultados_consulta.py

@@ -61,10 +61,11 @@ def retrieve_index(data,cosine_scores, document_index, topn=10):
     data = data.set_index(document_index)
     data['scores'] = cosine_scores
     df_top_scores=data.reset_index().sort_values('scores',ascending=False).head(topn)
-    df_top_scores=df_top_scores[df_top_scores['scores'] > 0]
+    cutoff=np.average(df_top_scores['scores'])+0.75*np.std(df_top_scores['scores'])
+    df_top_scores=df_top_scores[df_top_scores['scores'] > cutoff]
     st.write(df_top_scores)
 
-    min=np.average(df_top_scores['scores'])+0.75*np.std(df_top_scores['scores'])
+   
     
     return df_top_scores.index # Dataframe original con los scores 
                                # del coseno en una columna nueva