brunorosilva commited on
Commit
4f82987
1 Parent(s): 3fcf4ec

feat: remove tmp file

Browse files
Files changed (1) hide show
  1. makeitsports_bot/models/predict.py +1 -0
makeitsports_bot/models/predict.py CHANGED
@@ -26,4 +26,5 @@ def predict(img: Image):
26
  for idx, _ in zip(indices[0], distances[0]):
27
  inv_tensor = inverse_transform_img(wikiart_dataset[idx][1]).cpu().numpy()
28
  results.append(inv_tensor)
 
29
  return results
 
26
  for idx, _ in zip(indices[0], distances[0]):
27
  inv_tensor = inverse_transform_img(wikiart_dataset[idx][1]).cpu().numpy()
28
  results.append(inv_tensor)
29
+ os.remove(tmp_img_path)
30
  return results