update url

rosettafold_pymol.py

@@ -2,6 +2,7 @@ from pymol import cmd
 import requests
 
 import os
+import time
 
 # from gradio_client import Client
 
@@ -133,10 +134,14 @@ def query_rosettafold2(
     num_models:
     Number of models Default:0
     """
-    if os.path.exists("/home/user/app/"):
-        url = "https://simonduerr-rosettafold2.hf.space"
-    else:
+
+    # check if server is running on localhost
+
+    try:
+        requests.get("http://localhost:7860")
         url = "http://localhost:7860"
+    except requests.exceptions.ConnectionError:
+        url = "https://simonduerr-rosettafold2.hf.space"
 
     response = requests.post(
         url + "/run/rosettafold2/",
@@ -162,7 +167,7 @@ def query_rosettafold2(
     try:
         data = response["data"]
     except KeyError:
-        print(response["error"])
+        # print(response["error"])
         return None
     with open(f"{jobname}.pdb", "w") as out:
         out.writelines(data)