Update app.py
Browse files
app.py
CHANGED
@@ -425,10 +425,6 @@ def process_diary(day, date, total_people, total_machinery, machinery_types, act
|
|
425 |
return [day, date, "Error processing media", "Error processing media", "Error processing media", "Error processing media", None]
|
426 |
|
427 |
finally:
|
428 |
-
# Clean up GPU memory
|
429 |
-
if torch.cuda.is_available():
|
430 |
-
torch.cuda.empty_cache()
|
431 |
-
gc.collect()
|
432 |
# Remove temporary file if downloaded from Azure
|
433 |
if media_source == "Azure Blob" and media_path and os.path.exists(media_path):
|
434 |
try:
|
|
|
425 |
return [day, date, "Error processing media", "Error processing media", "Error processing media", "Error processing media", None]
|
426 |
|
427 |
finally:
|
|
|
|
|
|
|
|
|
428 |
# Remove temporary file if downloaded from Azure
|
429 |
if media_source == "Azure Blob" and media_path and os.path.exists(media_path):
|
430 |
try:
|