Update ssh

serve/gradio_web.py

@@ -59,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()

serve/upload.py

@@ -26,6 +26,7 @@ def get_image_from_url(image_url):
 
 def get_random_mscoco_prompt():
     global sftp_client
+    create_ssh_client(SSH_SERVER, SSH_PORT, SSH_USER, SSH_PASSWORD)
     num = random.randint(0, 2999)
     file = "{}.txt".format(num)