Update app.py

app.py

@@ -193,21 +193,27 @@ class ResNetDatasetCreator:
         print(f"Saved {len(self.temp_processed_dogs)} dogs to permanent database")
     
     def clear_temp_data(self):
-        """Clear all temporary data for new video"""
+        """Clear all temporary data for new video and free memory."""
         # Clear temp directory
         if self.temp_dir.exists():
             shutil.rmtree(self.temp_dir)
         self.temp_dir.mkdir()
-        
+    
         # Clear temp session data
         self.current_video_path = None
         self.current_session = None
         self.temp_processed_dogs = {}
-        
+    
         # Reset ReID (clears in-memory dogs)
         self.reid.reset_all()
         
-        print("Temporary data cleared for new video")
+        # 👇 ADD THESE TWO LINES FOR MEMORY CLEANUP
+        gc.collect()
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
+    
+        print("Temporary data cleared and memory released.")
+
     
     def clear_all_permanent_data(self):
         """Clear entire permanent database"""