Add rules

model/matchmaker.py

@@ -81,6 +81,9 @@ def matchmaker(num_players, k_group=4):
     
     ratings, comparison_counts, total_comparisons = load_json_via_sftp()
 
+    ratings = ratings[:num_players]
+    comparison_counts = comparison_counts[:num_players, :num_players]
+
     # Randomly select a player
     # selected_player = np.random.randint(0, num_players)
     selected_player = np.argmin(comparison_counts.sum(axis=1))

model/models/other_api_models.py

@@ -9,7 +9,8 @@ class OtherModel():
         self.model_name = model_name
         self.model_type = model_type
         self.url = "https://www.xdai.online/mj/submit/imagine"
-        self.key = os.environ.get('MIDJOURNEY_KEY')
+        # self.key = os.environ.get('MIDJOURNEY_KEY')
+        self.key = 'sk-yhIAe209CeWmk7ueC5D673A2D6164eE5919656Ea989913Fb'
         self.get_url = "https://www.xdai.online/mj/image/"
         self.repeat_num = 5