Spaces:
Runtime error
Runtime error
| import os | |
| import spaces | |
| from demo.app import EXAMPLES_IMAGE_DIR, _style_kwargs, build_demo | |
| from wevisone_vl import WeVisOneVL | |
| MODEL_ID = os.environ.get("WEVISONE_MODEL", "tencent/WeVisOne-VL-4B") | |
| engine = WeVisOneVL(MODEL_ID, backend="transformers", device="cuda") | |
| engine.predict = spaces.GPU(duration=120)(engine.predict) | |
| demo = build_demo(engine) | |
| demo.queue() | |
| demo.launch(allowed_paths=[str(EXAMPLES_IMAGE_DIR)], **_style_kwargs("launch")) | |