some refactor

src/hub.py

@@ -11,7 +11,8 @@ def create_space_with_content(
     space_card: str,
     token: str,
 ):
-    logging.info(f"Creating space with content: {repo_id} on file {html_file_path}")
+    logging.info(f"Creating space {repo_id}")
+
     api.create_repo(
         repo_id=repo_id,
         repo_type="space",
@@ -20,32 +21,29 @@ def create_space_with_content(
         token=token,
         space_sdk="static",
     )
-
     SpaceCard(content=space_card.format(dataset_id=dataset_id)).push_to_hub(
         repo_id=repo_id, repo_type="space", token=token
     )
 
+    upload_file(api, html_file_path, "index.html", repo_id, token)
+    upload_file(api, plot_file_path, "static_plot.png", repo_id, token)
+
+    logging.info("Space creation done")
+    return repo_id
+
+
+def upload_file(
+    api: HfApi,
+    file_path: str,
+    path_in_repo: str,
+    repo_id: str,
+    token: str,
+    repo_type: str = "space",
+):
     api.upload_file(
-        path_or_fileobj=html_file_path,
-        path_in_repo="index.html",
-        repo_type="space",
+        path_or_fileobj=file_path,
+        path_in_repo=path_in_repo,
+        repo_type=repo_type,
         repo_id=repo_id,
         token=token,
     )
-
-    logging.info(f"Pushing file to hub: {dataset_id} on file {plot_file_path}")
-
-    try:
-        logging.info(f"About to push {plot_file_path} - {dataset_id}")
-        api.upload_file(
-            path_or_fileobj=plot_file_path,
-            path_in_repo="static_plot.png",
-            repo_id=repo_id,
-            repo_type="space",
-        )
-    except Exception as e:
-        logging.info("Failed to push file", e)
-        raise
-
-    logging.info(f"Space creation done")
-    return repo_id

src/templates.py

@@ -20,7 +20,7 @@ Based on the information about the topic above, please create a short label of t
 [/INST]
 """
 
-REPRESENTATION_PROMPT = SYSTEM_PROMPT + EXAMPLE_PROMPT + MAIN_PROMPT
+REPRESENTATION_PROMPT = f"{SYSTEM_PROMPT}{EXAMPLE_PROMPT}{MAIN_PROMPT}"
 
 SPACE_REPO_CARD_CONTENT = """
 ---