testbot commited on
Commit
526c343
β€’
1 Parent(s): 02d2aee

even more explciit

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  from pathlib import Path
3
  from tempfile import TemporaryDirectory
4
 
@@ -129,7 +130,7 @@ def run(
129
  yield _log(
130
  f"### Success πŸ”₯\n\nYay! This model was successfully converted! Make sure to let the repo owner know about it and review your PR. You might need to complete the PR manually, especially to add information in the model card."
131
  )
132
- model_path.unlink(missing_ok=True)
133
  _delete_cache()
134
  return
135
  except Exception as e:
 
1
  import os
2
+ import shutil
3
  from pathlib import Path
4
  from tempfile import TemporaryDirectory
5
 
 
130
  yield _log(
131
  f"### Success πŸ”₯\n\nYay! This model was successfully converted! Make sure to let the repo owner know about it and review your PR. You might need to complete the PR manually, especially to add information in the model card."
132
  )
133
+ shutil.rmtree(model_path.parent)
134
  _delete_cache()
135
  return
136
  except Exception as e: