Spaces:
Running
on
A100
Running
on
A100
next frame if image is None
Browse files- app_init.py +2 -5
app_init.py
CHANGED
@@ -121,17 +121,14 @@ def init_app(app: FastAPI, user_data: UserData, args: Args, pipeline):
|
|
121 |
last_params = params
|
122 |
image = pipeline.predict(params)
|
123 |
if image is None:
|
|
|
124 |
continue
|
125 |
frame = pil_to_frame(image)
|
126 |
yield frame
|
127 |
# https://bugs.chromium.org/p/chromium/issues/detail?id=1250396
|
128 |
if not is_firefox(request.headers["user-agent"]):
|
129 |
yield frame
|
130 |
-
await websocket.send_json(
|
131 |
-
{
|
132 |
-
"status": "send_frame",
|
133 |
-
}
|
134 |
-
)
|
135 |
|
136 |
return StreamingResponse(
|
137 |
generate(),
|
|
|
121 |
last_params = params
|
122 |
image = pipeline.predict(params)
|
123 |
if image is None:
|
124 |
+
await websocket.send_json({"status": "send_frame"})
|
125 |
continue
|
126 |
frame = pil_to_frame(image)
|
127 |
yield frame
|
128 |
# https://bugs.chromium.org/p/chromium/issues/detail?id=1250396
|
129 |
if not is_firefox(request.headers["user-agent"]):
|
130 |
yield frame
|
131 |
+
await websocket.send_json({"status": "send_frame"})
|
|
|
|
|
|
|
|
|
132 |
|
133 |
return StreamingResponse(
|
134 |
generate(),
|