Update app.py

app.py

@@ -70,7 +70,7 @@ def aggregate(list_of_hits):
 
     preferences = []
     for result in list_of_hits:
-        preferences.append([doc["rank"] for doc in result])
+        preferences.append([doc["rank"] - 1 for doc in result])
 
     preferences = np.array(preferences)
     # y_optimal = KemenyOptimalAggregator().aggregate(preferences)
@@ -78,7 +78,7 @@ def aggregate(list_of_hits):
 
     rank2doc = {}
     for doc in list_of_hits[0]:
-        rank2doc[doc["rank"]] = doc
+        rank2doc[doc["rank"] - 1] = doc
 
     print("preferences: ", preferences.shape, preferences[0])
     print("rank2doc:", rank2doc.keys())