update
Browse files
app.py
CHANGED
|
@@ -9,6 +9,7 @@ import torch
|
|
| 9 |
from tqdm import trange
|
| 10 |
import imageio
|
| 11 |
import requests
|
|
|
|
| 12 |
|
| 13 |
print('load!', flush=True)
|
| 14 |
checkpoint = 'clevr.pth'
|
|
@@ -267,6 +268,10 @@ with gr.Blocks() as demo:
|
|
| 267 |
|
| 268 |
# btn.click(fn=predict, inputs=num_frames, outputs=gallery, postprocess=False)
|
| 269 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 270 |
demo.queue()
|
| 271 |
-
demo.launch(server_name='0.0.0.0', server_port=
|
| 272 |
# demo.launch(server_name='0.0.0.0', server_port=7860, debug=True, show_error=True)
|
|
|
|
| 9 |
from tqdm import trange
|
| 10 |
import imageio
|
| 11 |
import requests
|
| 12 |
+
import argparse
|
| 13 |
|
| 14 |
print('load!', flush=True)
|
| 15 |
checkpoint = 'clevr.pth'
|
|
|
|
| 268 |
|
| 269 |
# btn.click(fn=predict, inputs=num_frames, outputs=gallery, postprocess=False)
|
| 270 |
|
| 271 |
+
parser = argparse.ArgumentParser()
|
| 272 |
+
parser.add_argument('--port', type=int, help='The port number', default=7860)
|
| 273 |
+
args = parser.parse_args()
|
| 274 |
+
|
| 275 |
demo.queue()
|
| 276 |
+
demo.launch(server_name='0.0.0.0', server_port=args.port, debug=True, show_error=True)
|
| 277 |
# demo.launch(server_name='0.0.0.0', server_port=7860, debug=True, show_error=True)
|