resolve the dependency issue of HF

app.py

@@ -5,6 +5,16 @@ import numpy as np
 import torch
 from PIL import Image
 
+import sys
+import subprocess
+
+try:
+    subprocess.check_call(["/bin/bash", "/install_package.sh"])
+except subprocess.CalledProcessError as e:
+    print(f"Script failed with return code {e.returncode}", file=sys.stderr)
+    sys.exit(e.returncode)
+
+
 from uniflowmatch.models.ufm import (
     UniFlowMatchClassificationRefinement,
     UniFlowMatchConfidence,

install_package.sh

@@ -0,0 +1,8 @@
+#/bin/bash
+
+cd UniCeption
+pip install -e .
+cd ..
+
+# install uniflowmatch
+pip install -e .

requirements.txt

@@ -1,13 +1,20 @@
-torch
-torchvision
-torchaudio
-numpy
-matplotlib
-opencv-python
-flow_vis
-huggingface_hub
+black
 einops
+flow_vis
 gradio
-
--e /home/user/app
--e /home/user/app/UniCeption
+huggingface_hub
+isort
+jaxtyping
+matplotlib
+minio
+numpy
+opencv-python
+Pillow
+pre-commit
+pytest
+rerun-sdk
+scikit-learn
+timm
+torch
+torchaudio
+torchvision