Update inference_from_video.py

inference_from_video.py

@@ -123,11 +123,16 @@ def main():
     model.eval()
 
     # Load Trained Weight #
-    if args.model.endswith(".pt") or args.model.endswith(".bin"):
+    try:
+        if args.model.endswith(".pt") or args.model.endswith(".bin"):
+            model.load_state_dict(torch.load(args.model, map_location=device), strict=False)
+        else:
+            from safetensors.torch import load_model
+            load_model(model, args.model, strict=False)
+    except OSError as e:
+        print(f"Error loading model with safetensors: {e}")
+        print("Falling back to torch.load")
         model.load_state_dict(torch.load(args.model, map_location=device), strict=False)
-    else:
-        from safetensors.torch import load_model
-        load_model(model, args.model, strict=False)
         
     model.to(device)