Update app.py

app.py

@@ -238,9 +238,19 @@ def _get_face_embedding(
             
             emb = emb / np.linalg.norm(emb)
             embeddings.append(emb.astype(float).tolist())
+
+        try:
+            del mtcnn
+            del facenet
+        except:
+            pass
+
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
+            torch.cuda.ipc_collect()
         
         return embeddings
-
+    
     except Exception as e:
         print(f"Face embedding failed: {e}")
         return None