Update app.py

app.py

@@ -10,53 +10,54 @@ from huggingface_hub import snapshot_download
 import sys, pathlib
 import os
 os.environ["OPENAI_API_KEY"] = "test"
-
-# 1) Download the repo to a local cache dir
-print("Downloading remote vine repo...")
-repo_dir = snapshot_download(repo_id="KevinX-Penn28/testing", revision="main")
-
-# 2) Register the snapshot as an importable package
-VINE_PACKAGE = "vine_remote_repo"
-
-# Drop stale modules in case the script reloads
-for module_name in list(sys.modules):
-    if module_name == VINE_PACKAGE or module_name.startswith(f"{VINE_PACKAGE}."):
-        del sys.modules[module_name]
-
-print("Dropped stale modules and registering vine package...")
-
-package_spec = importlib.util.spec_from_file_location(
-    VINE_PACKAGE,
-    Path(repo_dir) / "__init__.py",
-    submodule_search_locations=[str(repo_dir)],
-)
-if not package_spec or not package_spec.loader:
-    raise ImportError(f"Cannot create package spec for {VINE_PACKAGE} at {repo_dir}")
-
-print("Created package spec, loading module...")
-
-package_module = importlib.util.module_from_spec(package_spec)
-sys.modules[VINE_PACKAGE] = package_module
-
-try:
-    print("Executing module...")
-    package_spec.loader.exec_module(package_module)
-    print("Module executed successfully!")
-except Exception as e:
-    print(f"ERROR during module execution: {e}")
-    import traceback
-    traceback.print_exc()
-    raise
-
-# 3) Import and use via the registered package
-print("Importing vine modules...")
-vine_config_module = importlib.import_module(f"{VINE_PACKAGE}.vine_config")
-vine_model_module = importlib.import_module(f"{VINE_PACKAGE}.vine_model")
-vine_pipeline_module = importlib.import_module(f"{VINE_PACKAGE}.vine_pipeline")
-
-VineConfig = vine_config_module.VineConfig              # your config class
-VineModel = vine_model_module.VineModel                 # your model class
-VinePipeline = vine_pipeline_module.VinePipeline        
+from vine_hf import VineConfig, VineModel, VinePipeline
+
+# # 1) Download the repo to a local cache dir
+# print("Downloading remote vine repo...")
+# repo_dir = snapshot_download(repo_id="KevinX-Penn28/testing", revision="main")
+
+# # 2) Register the snapshot as an importable package
+# VINE_PACKAGE = "vine_remote_repo"
+
+# # Drop stale modules in case the script reloads
+# for module_name in list(sys.modules):
+#     if module_name == VINE_PACKAGE or module_name.startswith(f"{VINE_PACKAGE}."):
+#         del sys.modules[module_name]
+
+# print("Dropped stale modules and registering vine package...")
+
+# package_spec = importlib.util.spec_from_file_location(
+#     VINE_PACKAGE,
+#     Path(repo_dir) / "__init__.py",
+#     submodule_search_locations=[str(repo_dir)],
+# )
+# if not package_spec or not package_spec.loader:
+#     raise ImportError(f"Cannot create package spec for {VINE_PACKAGE} at {repo_dir}")
+
+# print("Created package spec, loading module...")
+
+# package_module = importlib.util.module_from_spec(package_spec)
+# sys.modules[VINE_PACKAGE] = package_module
+
+# try:
+#     print("Executing module...")
+#     package_spec.loader.exec_module(package_module)
+#     print("Module executed successfully!")
+# except Exception as e:
+#     print(f"ERROR during module execution: {e}")
+#     import traceback
+#     traceback.print_exc()
+#     raise
+
+# # 3) Import and use via the registered package
+# print("Importing vine modules...")
+# vine_config_module = importlib.import_module(f"{VINE_PACKAGE}.vine_config")
+# vine_model_module = importlib.import_module(f"{VINE_PACKAGE}.vine_model")
+# vine_pipeline_module = importlib.import_module(f"{VINE_PACKAGE}.vine_pipeline")
+
+# VineConfig = vine_config_module.VineConfig              # your config class
+# VineModel = vine_model_module.VineModel                 # your model class
+# VinePipeline = vine_pipeline_module.VinePipeline        
 
 current_dir = Path(__file__).resolve().parent
 sam_config_path = "/" + str(Path(current_dir) / "sam2_hiera_t.yaml")
@@ -65,14 +66,14 @@ gd_config_path = "/" + str(Path(current_dir) / "GroundingDINO_SwinT_OGC.py")
 gd_checkpoint_path = "/" + str(Path(current_dir) / "groundingdino_swint_ogc.pth")
 visualization_dir = "/" + str(Path(current_dir) / "outputs")    
 
-# current_dir = Path.cwd()
-# sam_config_path = "/" + str(current_dir / "sam2_hiera_t.yaml")
-# sam_checkpoint_path = "/" + str(current_dir / "sam2_hiera_tiny.pt")
-# gd_config_path = "/" + str(current_dir / "GroundingDINO_SwinT_OGC.py")
-# gd_checkpoint_path = "/" + str(current_dir / "groundingdino_swint_ogc.pth")
-# visualization_dir = "/" + str(current_dir / "outputs")
+# # current_dir = Path.cwd()
+# # sam_config_path = "/" + str(current_dir / "sam2_hiera_t.yaml")
+# # sam_checkpoint_path = "/" + str(current_dir / "sam2_hiera_tiny.pt")
+# # gd_config_path = "/" + str(current_dir / "GroundingDINO_SwinT_OGC.py")
+# # gd_checkpoint_path = "/" + str(current_dir / "groundingdino_swint_ogc.pth")
+# # visualization_dir = "/" + str(current_dir / "outputs")
 
-print(f"Setting up paths done: {sam_config_path}, {sam_checkpoint_path}, {gd_config_path}, {gd_checkpoint_path}")
+# print(f"Setting up paths done: {sam_config_path}, {sam_checkpoint_path}, {gd_config_path}, {gd_checkpoint_path}")
 
 def process_video(video_file, categorical_keywords, unary_keywords, binary_keywords, object_pairs, output_fps):