Update ssh

serve/upload.py

@@ -26,17 +26,14 @@ def get_image_from_url(image_url):
 
 def get_random_mscoco_prompt():
     global sftp_client
-    files = sftp_client.listdir(SSH_MSCOCO)
-    txt_files = [file for file in files if file.endswith('.txt')]
+    num = random.randint(0, 2999)
+    file = "{}.txt".format(num)
 
-    selected_files = random.sample(txt_files, 1) # get one prompt
-
-    for file in selected_files:
-        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")
+    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