30 lines
652 B
Python
30 lines
652 B
Python
|
import os
|
||
|
|
||
|
def script_method(fn, _rcb=None):
|
||
|
return fn
|
||
|
|
||
|
def script(obj, optimize=True, _frames_up=0, _rcb=None):
|
||
|
return obj
|
||
|
|
||
|
import torch.jit
|
||
|
torch.jit.script_method = script_method
|
||
|
torch.jit.script = script
|
||
|
|
||
|
import multiprocessing
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
multiprocessing.freeze_support()
|
||
|
try:
|
||
|
from onnx_web.main import main
|
||
|
app, pool = main()
|
||
|
print("starting workers")
|
||
|
pool.start()
|
||
|
print("starting flask")
|
||
|
app.run("0.0.0.0", 5000, debug=False)
|
||
|
input("press the any key")
|
||
|
pool.join()
|
||
|
except Exception as e:
|
||
|
print(e)
|
||
|
finally:
|
||
|
os.system("pause")
|