Update ssh

app.py

@@ -4,10 +4,7 @@ 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, 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
+from serve.constants import SERVER_PORT, ROOT_PATH, ELO_RESULTS_DIR
 
 def build_combine_demo(models, elo_results_file, leaderboard_table_file):
         # gr.themes.Default(),
@@ -53,11 +50,6 @@ def load_elo_results(elo_results_dir):
             
     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

serve/gradio_web.py

@@ -36,6 +36,14 @@ from .Ksort import (
 
 from functools import partial
 from .upload import get_random_mscoco_prompt
+from serve.constants import 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 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)
 
 def build_side_by_side_ui_anony(models):
     notice_markdown = """
@@ -51,7 +59,7 @@ def build_side_by_side_ui_anony(models):
 """
 
     model_list = models.model_ig_list
-
+    set_ssh()
     state0 = gr.State()
     state1 = gr.State()
     state2 = gr.State()
@@ -264,7 +272,7 @@ def build_side_by_side_ui_anony(models):
     # Top2_btn.click(reset_level, inputs=[Top2_text], outputs=[vote_level])
     # Top3_btn.click(reset_level, inputs=[Top3_text], outputs=[vote_level])
     # Top4_btn.click(reset_level, inputs=[Top4_text], outputs=[vote_level])
-    vote_mode = gr.Textbox(value="Best", visible=False, interactive=False)
+    vote_mode = gr.Textbox(value="Rank", visible=False, interactive=False)
     right_vote_text = gr.Textbox(value="wrong", visible=False, interactive=False)
 
     textbox.submit(

serve/upload.py

@@ -69,4 +69,5 @@ def upload_ssh_all(states, output_dir, data, data_path):
     json_data = json.dumps(data, indent=4)
     with io.BytesIO(json_data.encode('utf-8')) as json_byte_stream:
         sftp_client.putfo(json_byte_stream, data_path)
-    print(f"Successfully uploaded JSON data to {data_path}")
+    print(f"Successfully uploaded JSON data to {data_path}")
+    create_ssh_client(SSH_SERVER, SSH_PORT, SSH_USER, SSH_PASSWORD)