Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -113,6 +113,7 @@ def update_features(
|
|
| 113 |
num_patches,
|
| 114 |
use_dummy
|
| 115 |
):
|
|
|
|
| 116 |
"""
|
| 117 |
Given a PIL image, returns:
|
| 118 |
1) the same PIL image (so it can be displayed)
|
|
|
|
| 113 |
num_patches,
|
| 114 |
use_dummy
|
| 115 |
):
|
| 116 |
+
torch.cuda.empty_cache()
|
| 117 |
"""
|
| 118 |
Given a PIL image, returns:
|
| 119 |
1) the same PIL image (so it can be displayed)
|