odunkel commited on
Commit
98aae95
·
verified ·
1 Parent(s): c74a8c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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)