Upload app_federal.py

app_federal.py

@@ -330,11 +330,11 @@ def run_retrieval(query):
     #print (out_dict[:50])
     res = {"result_type":"chatgpt_reranking"}
     res["query"] = query
-    res["input_reranking"] = [i["index"]] for i in out_dict
+    res["input_reranking"] = [i["index"] for i in out_dict]
     res["scores_input_reranking"] = [i["NV_score"] for i in out_dict]
     out_dict = rerank_with_chatGPT(query, out_dict)[:NUM_RESULTS]
 
-    res["output_reranking"] = [i["index"]] for i in out_dict
+    res["output_reranking"] = [i["index"] for i in out_dict]
     res["scores_output_reranking"] = [i["NV_score"] for i in out_dict]
 
     # is that already good?