hysts HF Staff commited on
Commit
8224854
·
1 Parent(s): a594ef6
Files changed (1) hide show
  1. app.py +7 -8
app.py CHANGED
@@ -336,14 +336,13 @@ def _get_prompt_for_obj(state: AppState, obj_id: int) -> str | None:
336
  return frame_texts[obj_id].strip()
337
 
338
  # Priority 2: Check inference session mapping
339
- if state.inference_session is not None:
340
- if (
341
- hasattr(state.inference_session, "obj_id_to_prompt_id")
342
- and obj_id in state.inference_session.obj_id_to_prompt_id
343
- ):
344
- prompt_id = state.inference_session.obj_id_to_prompt_id[obj_id]
345
- if hasattr(state.inference_session, "prompts") and prompt_id in state.inference_session.prompts:
346
- return state.inference_session.prompts[prompt_id].strip()
347
 
348
  return None
349
 
 
336
  return frame_texts[obj_id].strip()
337
 
338
  # Priority 2: Check inference session mapping
339
+ if state.inference_session is not None and (
340
+ hasattr(state.inference_session, "obj_id_to_prompt_id")
341
+ and obj_id in state.inference_session.obj_id_to_prompt_id
342
+ ):
343
+ prompt_id = state.inference_session.obj_id_to_prompt_id[obj_id]
344
+ if hasattr(state.inference_session, "prompts") and prompt_id in state.inference_session.prompts:
345
+ return state.inference_session.prompts[prompt_id].strip()
 
346
 
347
  return None
348