CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-05-02 06:15:09 +02:00
Code Issues Packages Projects Releases Wiki Activity

remove while True and websocket code

Browse Source
This commit is contained in:
harisreedhar
2026-02-22 14:22:46 +05:30
committed by henryruhs
parent 7957640f61
commit 6152ecc973
1 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/apis/endpoints/process.py
+8 -9
View File
@@ -20,19 +20,18 @@ async def websocket_process_image(websocket : WebSocket) -> None:
if source_paths:
try:
while True:
image_bytes = await websocket.receive_bytes()
target_vision_frame = cv2.imdecode(numpy.frombuffer(image_bytes, numpy.uint8), cv2.IMREAD_COLOR)
image_bytes = await websocket.receive_bytes()
target_vision_frame = cv2.imdecode(numpy.frombuffer(image_bytes, numpy.uint8), cv2.IMREAD_COLOR)
if numpy.any(target_vision_frame):
temp_vision_frame = process_stream_frame(target_vision_frame)
is_success, output_vision_frame = cv2.imencode('.jpg', temp_vision_frame)
if numpy.any(target_vision_frame):
temp_vision_frame = process_stream_frame(target_vision_frame)
is_success, output_vision_frame = cv2.imencode('.jpg', temp_vision_frame)
if is_success:
await websocket.send_bytes(output_vision_frame.tobytes())
if is_success:
await websocket.send_bytes(output_vision_frame.tobytes())
except (WebSocketDisconnect, OSError):
pass
return
await websocket.close(code = 1008)
await websocket.close()