Add more Giskard hub info for uploading

text_classification_ui_helpers.py

@@ -24,7 +24,10 @@ MAX_FEATURES = 20
 HF_REPO_ID = "HF_REPO_ID"
 HF_SPACE_ID = "SPACE_ID"
 HF_WRITE_TOKEN = "HF_WRITE_TOKEN"
-HF_GSK_HUB_KEY = "GSK_HUB_KEY"
+HF_GSK_HUB_URL = "GSK_HUB_URL"
+HF_GSK_HUB_PROJECT_KEY = "GSK_HUB_PROJECT_KEY"
+HF_GSK_HUB_KEY = "GSK_API_KEY"
+HF_GSK_HUB_HF_TOKEN = "GSK_HF_TOKEN"
 HF_GSK_HUB_UNLOCK_TOKEN = "GSK_HUB_UNLOCK_TOKEN"
 
 
@@ -250,6 +253,15 @@ def try_submit(m_id, d_id, config, split, local, uid):
         if os.environ.get(HF_GSK_HUB_KEY):
             command.append("--giskard_hub_api_key")
             command.append(os.environ.get(HF_GSK_HUB_KEY))
+            if os.environ.get(HF_GSK_HUB_URL):
+                command.append("--giskard_hub_url")
+                command.append(os.environ.get(HF_GSK_HUB_URL))
+            if os.environ.get(HF_GSK_HUB_PROJECT_KEY):
+                command.append("--giskard_hub_project_key")
+                command.append(os.environ.get(HF_GSK_HUB_PROJECT_KEY))
+            if os.environ.get(HF_GSK_HUB_HF_TOKEN):
+                command.append("--giskard_hub_hf_token")
+                command.append(os.environ.get(HF_GSK_HUB_HF_TOKEN))
             if os.environ.get(HF_GSK_HUB_UNLOCK_TOKEN):
                 command.append("--giskard_hub_unlock_token")
                 command.append(os.environ.get(HF_GSK_HUB_UNLOCK_TOKEN))