Spaces:
Running
on
Zero
Running
on
Zero
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)
|