amirDev commited on
Commit
91ef522
·
1 Parent(s): dd5c374

Update inference_flask.py

Browse files
Files changed (1) hide show
  1. inference_flask.py +1 -1
inference_flask.py CHANGED
@@ -51,7 +51,7 @@ def load_model():
51
  print('Downloading VGG!')
52
  url = model_urls['vgg16_bn']
53
  r = requests.get(url, allow_redirects=True)
54
- with open(model_paths['vgg16_bn'], 'w') as f:
55
  f.write(r.content)
56
  print('Finished Downloading')
57
  parser = argparse.ArgumentParser('P2PNet evaluation script', parents=[get_args_parser()])
 
51
  print('Downloading VGG!')
52
  url = model_urls['vgg16_bn']
53
  r = requests.get(url, allow_redirects=True)
54
+ with open(model_paths['vgg16_bn'], 'wb') as f:
55
  f.write(r.content)
56
  print('Finished Downloading')
57
  parser = argparse.ArgumentParser('P2PNet evaluation script', parents=[get_args_parser()])