Spaces:
Sleeping
Sleeping
KEYE commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,35 +5,28 @@ from loguru import logger
|
|
| 5 |
|
| 6 |
def setup_and_run():
|
| 7 |
repo_url = "https://github.com/renkexuan369/INF-MLLM.git"
|
| 8 |
-
|
|
|
|
| 9 |
|
| 10 |
-
|
| 11 |
-
|
| 12 |
-
# =========================================
|
| 13 |
-
if not os.path.exists(repo_dir):
|
| 14 |
-
logger.info(f"正在克隆仓库: {repo_url}")
|
| 15 |
subprocess.run(["git", "clone", repo_url], check=True)
|
| 16 |
else:
|
| 17 |
-
logger.info("
|
| 18 |
-
subprocess.run(["git", "pull"], cwd=
|
| 19 |
|
| 20 |
-
# 1. 注入安全环境变量
|
| 21 |
os.environ['GRADIO_SSR_MODE'] = "false"
|
| 22 |
|
| 23 |
-
# HF Space 默认暴露 7860 端口
|
| 24 |
os.environ["GRADIO_SERVER_PORT"] = "7860"
|
| 25 |
os.environ["GRADIO_SERVER_NAME"] = "0.0.0.0"
|
| 26 |
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
# =========================================
|
| 30 |
-
logger.info("准备启动 inf_gradio.py ...")
|
| 31 |
|
| 32 |
-
# 关键:必须设置 cwd=repo_dir,否则 Python 找不到 utils.py 和 prompts.py
|
| 33 |
try:
|
| 34 |
subprocess.run([sys.executable, "gradio_app.py"], cwd=repo_dir, check=True)
|
| 35 |
except subprocess.CalledProcessError as e:
|
| 36 |
-
logger.error(f"Gradio
|
| 37 |
|
| 38 |
if __name__ == "__main__":
|
| 39 |
setup_and_run()
|
|
|
|
| 5 |
|
| 6 |
def setup_and_run():
|
| 7 |
repo_url = "https://github.com/renkexuan369/INF-MLLM.git"
|
| 8 |
+
clone_dir = "INF-MLLM"
|
| 9 |
+
repo_dir = "INF-MLLM/Infinity-Parser2/infinity_parser2"
|
| 10 |
|
| 11 |
+
if not os.path.exists(clone_dir):
|
| 12 |
+
logger.info(f"clone: {repo_url}")
|
|
|
|
|
|
|
|
|
|
| 13 |
subprocess.run(["git", "clone", repo_url], check=True)
|
| 14 |
else:
|
| 15 |
+
logger.info("pulling...")
|
| 16 |
+
subprocess.run(["git", "pull"], cwd=clone_dir, check=True)
|
| 17 |
|
|
|
|
| 18 |
os.environ['GRADIO_SSR_MODE'] = "false"
|
| 19 |
|
|
|
|
| 20 |
os.environ["GRADIO_SERVER_PORT"] = "7860"
|
| 21 |
os.environ["GRADIO_SERVER_NAME"] = "0.0.0.0"
|
| 22 |
|
| 23 |
+
|
| 24 |
+
logger.info("run inf_gradio.py ...")
|
|
|
|
|
|
|
| 25 |
|
|
|
|
| 26 |
try:
|
| 27 |
subprocess.run([sys.executable, "gradio_app.py"], cwd=repo_dir, check=True)
|
| 28 |
except subprocess.CalledProcessError as e:
|
| 29 |
+
logger.error(f"Gradio exit: {e}")
|
| 30 |
|
| 31 |
if __name__ == "__main__":
|
| 32 |
setup_and_run()
|