From c59596e09645b66f89b87d7408c29bc7e52b7242 Mon Sep 17 00:00:00 2001
From: harisreedhar <h4harisreedhar.s.s@gmail.com>
Date: Fri, 20 Feb 2026 21:25:37 +0530
Subject: [PATCH] update exception

---
 facefusion/apis/endpoints/process.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/facefusion/apis/endpoints/process.py b/facefusion/apis/endpoints/process.py
index 7dc07937..c974615d 100644
--- a/facefusion/apis/endpoints/process.py
+++ b/facefusion/apis/endpoints/process.py
@@ -3,7 +3,7 @@ import uuid
 from time import time
 from typing import Optional
 
-from starlette.websockets import WebSocket
+from starlette.websockets import WebSocket, WebSocketDisconnect
 
 from facefusion import ffmpeg, process_manager, session_context, session_manager, state_manager
 from facefusion.apis.api_helper import get_sec_websocket_protocol
@@ -74,7 +74,7 @@ async def websocket_process_image(websocket : WebSocket) -> None:
 
 						remove_file(output_path)
 
-		except Exception:
+		except (WebSocketDisconnect, OSError):
 			pass
 		return