Update ssh

serve/leaderboard.py

@@ -134,7 +134,8 @@ def make_arena_leaderboard_md(elo_results):
     total_models = len(arena_df)
 
     leaderboard_md = f"""
-Total #models: **{total_models}**(anonymous). Total #votes: **{total_votes}**. Last updated: {last_updated}.
+Total #models: **{total_models}**(anonymous). Total #votes: **{total_votes}** (Equivalent to **{total_votes*6}** votes for one-on-one games). 
+Last updated: {last_updated}.
 """
 
     return leaderboard_md

serve/upload.py

@@ -41,19 +41,28 @@ def get_image_from_url(image_url):
     response.raise_for_status()  # success
     return Image.open(io.BytesIO(response.content))
 
+# def get_random_mscoco_prompt():
+#     global sftp_client
+#     if not is_connected():
+#         create_ssh_client(SSH_SERVER, SSH_PORT, SSH_USER, SSH_PASSWORD)
+#     num = random.randint(0, 2999)
+#     file = "{}.txt".format(num)
+
+#     remote_file_path = os.path.join(SSH_MSCOCO, file)
+#     with sftp_client.file(remote_file_path, 'r') as f:
+#         content = f.read().decode('utf-8')
+#         print(f"Content of {file}:")
+#         print("\n")
+#     return content
+
 def get_random_mscoco_prompt():
-    global sftp_client
-    if not is_connected():
-        create_ssh_client(SSH_SERVER, SSH_PORT, SSH_USER, SSH_PASSWORD)
-    num = random.randint(0, 2999)
-    file = "{}.txt".format(num)
 
-    remote_file_path = os.path.join(SSH_MSCOCO, file)
-    with sftp_client.file(remote_file_path, 'r') as f:
-        content = f.read().decode('utf-8')
-        print(f"Content of {file}:")
-        print("\n")
-    return content
+    file_path = './sum_prompt.txt'
+    with open(file_path, 'r') as file:
+        lines = file.readlines()
+
+    random_line = random.choice(lines).strip()
+    return random_line
 
 
 def create_remote_directory(remote_directory):
@@ -85,6 +94,7 @@ def upload_ssh_all(states, output_dir, data, data_path):
                 print("get url image")
                 image_list[i] = get_image_from_url(image_list[i])
             with io.BytesIO() as image_byte_stream:
+                image_list[i] = image_list[i].resize((512, 512), Image.ANTIALIAS)
                 image_list[i].save(image_byte_stream, format='JPEG')
                 image_byte_stream.seek(0)  
                 sftp.putfo(image_byte_stream, output_file_list[i])

serve/utils.py

@@ -19,22 +19,7 @@ no_change_btn = gr.update(value="No Change", interactive=True, visible=True)
 def build_about():
     about_markdown = f"""
 # About Us
-This is a project from TIGER Lab at University of Waterloo. 
 
-## Contributors:
-[Tianle Li](https://scholar.google.com/citations?user=g213g7YAAAAJ&hl=en), [Dongfu Jiang](https://jdf-prog.github.io/), [Yuansheng Ni](https://yuanshengni.github.io/), [Max Ku](https://kuwingfung.github.io/), [Shizhuo Sun](), [Richard Fan](https://www.linkedin.com/in/richard-fan2020/).
-
-## Contact:
-Email: [email protected] (Tianle Li)
-
-## Advisors
-[Wenhu Chen](https://wenhuchen.github.io/)
-
-## Sponsorship
-We are keep looking for sponsorship to support the arena project for the long term. Please contact us if you are interested in supporting this project.
-
-## Acknowledgment
-Our codebase is built upon <a href="https://github.com/lm-sys/FastChat" target="_blank">FastChat</a>, <a href="https://github.com/TIGER-AI-Lab/ImagenHub" target="_blank">ImagenHub</a> and <a href="https://github.com/TIGER-AI-Lab/VideoGenHub" target="_blank">VideoGenHub</a>.
 """
 
     gr.Markdown(about_markdown, elem_id="about_markdown")
@@ -43,7 +28,7 @@ Our codebase is built upon <a href="https://github.com/lm-sys/FastChat" target="
 acknowledgment_md = """
 ### Acknowledgment
 <div class="image-container">
-    <p> Our codebase is built upon <a href="https://github.com/lm-sys/FastChat" target="_blank">FastChat</a>, <a href="https://github.com/TIGER-AI-Lab/ImagenHub" target="_blank">ImagenHub</a> and <a href="https://github.com/TIGER-AI-Lab/VideoGenHub" target="_blank">VideoGenHub</a>.</p>
+    <p> Our codebase is built upon <a href="https://github.com/lm-sys/FastChat" target="_blank">FastChat</a>, <a href="https://github.com/TIGER-AI-Lab/ImagenHub" target="_blank">ImagenHub</a>.</p>
 </div>
 """
 block_css = """