Spaces:
Runtime error
Runtime error
brunorosilva
commited on
Commit
•
4f82987
1
Parent(s):
3fcf4ec
feat: remove tmp file
Browse files
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
|