assentian1970 commited on
Commit
587c721
·
verified ·
1 Parent(s): 74105d5

Update app.py

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