Spaces:
Build error
Build error
Update inference_flask.py
Browse files- 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'], '
|
| 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()])
|