update

constants.py

@@ -1,8 +1,6 @@
 from pathlib import Path
 import os
 
-print("here0")
-
 UNIDISC_DIR = Path(os.getenv("UNIDISC_DIR", Path(__file__).parent))
 LIB_DIR = UNIDISC_DIR / "third_party"
 CELEBV_DATA_DIR = Path(os.getenv("CELEBV_DATA_DIR", "/home/mprabhud/aswerdlo/repos/lib/CelebV-Text/downloaded_celebvtext"))
@@ -12,7 +10,6 @@ HF_TOKEN = os.getenv("HF_TOKEN", os.getenv("HF_HUB_DATASETS_TOKEN"))
 HF_DATASETS_CACHE = os.getenv("HF_DATASETS_CACHE", None)
 HF_CACHE_DIR = os.getenv("HF_HOME", None)
 
-print("here1")
 if HF_CACHE_DIR is not None:
     HF_CACHE_DIR = Path(HF_CACHE_DIR)
 elif HF_DATASETS_CACHE is not None:
@@ -24,5 +21,3 @@ try:
         CELEBV_DATA_DIR = SCRATCH_CELEBV_DATA_DIR
 except:
     print(f"Error setting CELEBV_DATA_DIR")
-
-print("here2")

demo/client.py

@@ -9,10 +9,14 @@ import numpy as np
 import io
 import json
 
+print(f"Finished import 0")
+
 SHOW_DEV_BUTTONS = True
-DEMO_DIR = Path("demo")
+DEMO_DIR = Path(__file__).parent
 ADD_DEV_FORM = True
 
+print(f"Finished import 1. Demo dir: {DEMO_DIR}")
+
 DEMOS = [
     {
         "name": "Dog",
@@ -656,5 +660,5 @@ def post(
 
     return output_content
 
-
+print(f"Before serve...")
 serve(port=5003)