update

app.py

@@ -31,8 +31,8 @@ from pathlib import Path
 import gradio as gr
 import pandas as pd
 from app_style import custom_theme, lighting_css
-# from embodied_gen.utils.tags import VERSION
-VERSION = "v0.1.8"
+from embodied_gen.utils.tags import VERSION
+
 try:
     from embodied_gen.utils.gpt_clients import GPT_CLIENT as gpt_client
 
@@ -50,6 +50,13 @@ except Exception as e:
 RUNNING_MODE = "hf_remote"  # local or hf_remote
 CSV_FILE = "dataset_index.csv"
 
+# Compatible with huggingface space zero GPU
+import spaces
+@spaces.GPU
+def fake_gpu_init():
+    pass
+fake_gpu_init()
+
 if RUNNING_MODE == "local":
     DATA_ROOT = "/horizon-bucket/robot_lab/datasets/embodiedgen/assets"
 elif RUNNING_MODE == "hf_remote":
@@ -724,9 +731,4 @@ with gr.Blocks(
 
 
 if __name__ == "__main__":
-    demo.launch(
-        # server_port=8088,
-        # allowed_paths=[
-        #     "/horizon-bucket/robot_lab/datasets/embodiedgen/assets"
-        # ],
-    )
+    demo.launch()

embodied_gen/utils/tags.py

@@ -0,0 +1 @@
+VERSION = "v0.1.8"