Spaces:
Running
on
Zero
Running
on
Zero
Clean up
Browse files
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 |
-
|
| 341 |
-
|
| 342 |
-
|
| 343 |
-
|
| 344 |
-
|
| 345 |
-
|
| 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 |
|