KEYE commited on
Commit
c45428a
·
verified ·
1 Parent(s): 3a61333

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -16
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
- repo_dir = "Infinity-Parser2/infinity_parser2"
 
9
 
10
- # =========================================
11
- # 1. 动态获取/拉取最新代码
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=repo_dir, check=True)
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
- # 2. 启动 Gradio 服务
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 服务异常退出: {e}")
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()