doublecheck inference_state["device"]

app.py

@@ -257,7 +257,8 @@ def segment_with_points(
     evt: gr.SelectData,
 ):
     predictor.to("cpu")
-    inference_state["device"] = predictor.device
+    if inference_state:
+        inference_state["device"] = predictor.device
     input_points.append(evt.index)
     print(f"TRACKING INPUT POINT: {input_points}")
 
@@ -344,7 +345,8 @@ def propagate_to_all(
         torch.backends.cudnn.allow_tf32 = True
     with torch.autocast(device_type="cuda", dtype=torch.bfloat16):
         predictor.to("cuda")
-        inference_state["device"] = predictor.device
+        if inference_state:
+            inference_state["device"] = predictor.device
 
         if len(input_points) == 0 or video_in is None or inference_state is None:
             return None