Sa-m commited on
Commit
5d5fa27
·
1 Parent(s): 3d15ada

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -43,9 +43,6 @@ if __name__ == "__main__":
43
  #serve(app, host="0.0.0.0", port=7860)
44
  '''
45
  if __name__ == "__main__":
46
- parser = argparse.ArgumentParser(description="Flask API exposing YOLOv5 model")
47
- parser.add_argument("--port", default=7860, type=int, help="port number")
48
- args = parser.parse_args()
49
-
50
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='best2.pt', force_reload=True) # force_reload to recache
51
- app.run(host="0.0.0.0", port=args.port,debug =True) # debug=True causes Restarting with stat
 
43
  #serve(app, host="0.0.0.0", port=7860)
44
  '''
45
  if __name__ == "__main__":
46
+
 
 
 
47
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='best2.pt', force_reload=True) # force_reload to recache
48
+ app.run(host="0.0.0.0", port=5000,debug =True) # debug=True causes Restarting with stat