lths commited on
Commit
4cf6c06
·
verified ·
1 Parent(s): c5d1a02

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -1,7 +1,11 @@
1
- # app.py(放在 D:\AI\VibeVoice)
2
  from demo.gradio_demo import main
 
3
 
4
  if __name__ == "__main__":
5
- # 如果你的 gradio_demo.py 不是 main() 入口,而是直接 launch(),
6
- # 可以改成:import demo.gradio_demo 以触发其顶层启动逻辑
 
 
 
7
  main()
 
1
+ # app.py
2
  from demo.gradio_demo import main
3
+ import sys
4
 
5
  if __name__ == "__main__":
6
+ # 强制覆写命令行参数,指定 Hub 仓库名
7
+ if "--model_path" not in sys.argv:
8
+ sys.argv += ["--model_path", "vibevoice/VibeVoice-1.5B"]
9
+ # 如果你的 Space 选了 GPU(T4),也可以加上:
10
+ # sys.argv += ["--device", "cuda"]
11
  main()