update app

load_app_data.py

@@ -1,14 +1,14 @@
 import os
 from huggingface_hub import snapshot_download
 
-hf_token = os.environ["HF_TOKEN"]
-app_data_repo = os.environ["APP_DATA_REPO"]
+HF_TOKEN = os.environ["HF_TOKEN"]
+APP_DATA_REPO = os.environ("APP_DATA_REPO")
 
 snapshot_download(
-    repo_id=f"sschellhammer/{app_data_repo}",
+    repo_id=f"{APP_DATA_REPO}",
     repo_type="dataset",
     local_dir="/app/potato",
-    token=hf_token
+    token=HF_TOKEN
 )
 
-print(f"downloaded app data from sschellhammer/{app_data_repo} to /app/potato")
+print(f"downloaded app data from {APP_DATA_REPO} to /app/potato")

load_existing_annotations.py

@@ -1,8 +1,8 @@
 import os
 from huggingface_hub import snapshot_download
 
-hf_token = os.environ["HF_TOKEN"]
-annotation_data_repo = os.environ["ANNOTATION_DATA_REPO"]
+HF_TOKEN = os.environ["HF_TOKEN"]
+ANNOTATION_DATA_REPO = os.environ("ANNOTATION_DATA_REPO")
 
-snapshot_download(repo_id=f"sschellhammer/{annotation_data_repo}", repo_type="dataset", local_dir="/app/potato/output", token=hf_token)
-print(f"downloaded existing annotations from sschellhammer/{annotation_data_repo} to /app/potato/output")
+snapshot_download(repo_id=f"{ANNOTATION_DATA_REPO}", repo_type="dataset", local_dir="/app/potato/output", token=HF_TOKEN)
+print(f"downloaded existing annotations from {ANNOTATION_DATA_REPO} to /app/potato/output")

start.py

@@ -1,11 +1,14 @@
 import os
 import subprocess
+from dotenv import load_dotenv
 from huggingface_hub import CommitScheduler
 
+load_dotenv("config.env")
+
 HF_TOKEN = os.environ["HF_TOKEN"]
 
 # Dataset repo to push to
-annotation_data_repo = "sschellhammer/"+os.environ["ANNOTATION_DATA_REPO"]
+ANNOTATION_DATA_REPO = os.environ("ANNOTATION_DATA_REPO")
 
 # Folder to monitor
 LOCAL_FOLDER = "/app/potato/output"
@@ -14,7 +17,7 @@ LOCAL_FOLDER = "/app/potato/output"
 INTERVAL = 1  # every 1 minute
 
 # Create CommitScheduler as context
-with CommitScheduler(repo_id=annotation_data_repo, token=HF_TOKEN, folder_path=LOCAL_FOLDER, repo_type="dataset", every=INTERVAL) as scheduler:
+with CommitScheduler(repo_id=ANNOTATION_DATA_REPO, token=HF_TOKEN, folder_path=LOCAL_FOLDER, repo_type="dataset", every=INTERVAL) as scheduler:
     print(f"Set up CommitScheduler listening to {LOCAL_FOLDER}")
 
     print("Load config")
@@ -25,5 +28,5 @@ with CommitScheduler(repo_id=annotation_data_repo, token=HF_TOKEN, folder_path=L
 
     print("Start Potato Server")
     os.chdir("potato")
-    subprocess.run(["potato", "start", "config.yaml", "-p", "7860", "--debug"])
+    subprocess.run(["potato", "start", "potato_config.yaml", "-p", "7860", "--debug"])
     

start.sh

@@ -2,6 +2,8 @@
 
 set -e  # exit on error
 
+source config.env
+
 echo "Pull latest version..."
 
 if [ -d "potato" ] && [ -d "potato/.git" ]; then
@@ -10,6 +12,11 @@ else
     git clone https://github.com/SEBSCHELLI/potato.git potato
 fi
 
+cd potato
+
+git checkout "$BRANCH"
+echo "Current branch: $BRANCH"
+
 pip install /app/potato
 
 echo "Starting Potato server..."