Update app.py
Browse files
app.py
CHANGED
|
@@ -290,8 +290,8 @@ def get_mask_sam_process(
|
|
| 290 |
torch.cuda.empty_cache()
|
| 291 |
|
| 292 |
# Assuming available_frames_to_check.value is a list
|
| 293 |
-
if working_frame not in available_frames_to_check
|
| 294 |
-
available_frames_to_check.
|
| 295 |
|
| 296 |
return "output_first_frame.jpg", frame_names, inference_state, gr.update(choices=available_frames_to_check.value, value=working_frame, visible=True)
|
| 297 |
|
|
|
|
| 290 |
torch.cuda.empty_cache()
|
| 291 |
|
| 292 |
# Assuming available_frames_to_check.value is a list
|
| 293 |
+
if working_frame not in available_frames_to_check:
|
| 294 |
+
available_frames_to_check.append(working_frame)
|
| 295 |
|
| 296 |
return "output_first_frame.jpg", frame_names, inference_state, gr.update(choices=available_frames_to_check.value, value=working_frame, visible=True)
|
| 297 |
|