radames commited on
Commit
7d67dc6
1 Parent(s): 7b4b3a3

next frame if image is None

Browse files
Files changed (1) hide show
  1. 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(),