Update ssh

app.py

@@ -4,7 +4,10 @@ from serve.gradio_web import *
 from serve.leaderboard import build_leaderboard_tab
 from model.model_manager import ModelManager
 from pathlib import Path
-from serve.constants import SERVER_PORT, ROOT_PATH, ELO_RESULTS_DIR
+from serve.constants import SERVER_PORT, ROOT_PATH, ELO_RESULTS_DIR, SSH_SERVER, SSH_PORT, SSH_USER, SSH_PASSWORD
+from serve.upload import get_random_mscoco_prompt, create_ssh_client
+from serve.update_skill import create_ssh_skill_client
+from model.matchmaker import create_ssh_matchmaker_client
 
 def build_combine_demo(models, elo_results_file, leaderboard_table_file):
         # gr.themes.Default(),
@@ -49,12 +52,18 @@ def load_elo_results(elo_results_dir):
             #     raise ValueError(f"Unknown file name: {file.name}")
             
     return elo_results_file, leaderboard_table_file
-    
+
+def set_ssh():
+    create_ssh_client(SSH_SERVER, SSH_PORT, SSH_USER, SSH_PASSWORD)
+    create_ssh_skill_client(SSH_SERVER, SSH_PORT, SSH_USER, SSH_PASSWORD)
+    create_ssh_matchmaker_client(SSH_SERVER, SSH_PORT, SSH_USER, SSH_PASSWORD)
+
 if __name__ == "__main__":
     server_port = int(SERVER_PORT)
     root_path = ROOT_PATH
     elo_results_dir = ELO_RESULTS_DIR
     models = ModelManager()
+    set_ssh()
     
     elo_results_file, leaderboard_table_file = load_elo_results(elo_results_dir)
     demo = build_combine_demo(models, elo_results_file, leaderboard_table_file)

serve/gradio_web.py

@@ -33,11 +33,9 @@ from .Ksort import (
     reset_vote_text,
     text_response_rank_igm,
 )
-from .upload import get_random_mscoco_prompt, create_ssh_client
-from .update_skill import create_ssh_skill_client
-from model.matchmaker import create_ssh_matchmaker_client
+
 from functools import partial
-from .constants import SSH_SERVER, SSH_PORT, SSH_USER, SSH_PASSWORD
+from .upload import get_random_mscoco_prompt
 
 def build_side_by_side_ui_anony(models):
     notice_markdown = """
@@ -53,9 +51,6 @@ def build_side_by_side_ui_anony(models):
 """
 
     model_list = models.model_ig_list
-    create_ssh_client(SSH_SERVER, SSH_PORT, SSH_USER, SSH_PASSWORD)
-    create_ssh_skill_client(SSH_SERVER, SSH_PORT, SSH_USER, SSH_PASSWORD)
-    create_ssh_matchmaker_client(SSH_SERVER, SSH_PORT, SSH_USER, SSH_PASSWORD)
 
     state0 = gr.State()
     state1 = gr.State()