Spaces:
Running
on
Zero
Running
on
Zero
lzyhha
commited on
Commit
·
307cfd5
1
Parent(s):
af44a4b
space
Browse files
app.py
CHANGED
@@ -524,7 +524,7 @@ def parse_args():
|
|
524 |
if __name__ == "__main__":
|
525 |
args = parse_args()
|
526 |
|
527 |
-
|
528 |
|
529 |
# Initialize model
|
530 |
model = VisualClozeModel(resolution=args.resolution, model_path=args.model_path, precision=args.precision)
|
@@ -533,5 +533,4 @@ if __name__ == "__main__":
|
|
533 |
demo = create_demo(model)
|
534 |
|
535 |
# Start Gradio server
|
536 |
-
|
537 |
-
demo.launch(share=False, server_port=10050, server_name="0.0.0.0")
|
|
|
524 |
if __name__ == "__main__":
|
525 |
args = parse_args()
|
526 |
|
527 |
+
snapshot_download(repo_id="VisualCloze/VisualCloze", repo_type="model", local_dir="checkpoints")
|
528 |
|
529 |
# Initialize model
|
530 |
model = VisualClozeModel(resolution=args.resolution, model_path=args.model_path, precision=args.precision)
|
|
|
533 |
demo = create_demo(model)
|
534 |
|
535 |
# Start Gradio server
|
536 |
+
demo.launch()
|
|