Merge pull request #205 from roboflow/fix-inference-fps

capjamesg · web-flow · commit 804f60edbc0d · 2023-12-01T16:57:23.000Z
Allow video inference `fps` value to pass through to API
diff --git a/roboflow/models/inference.py b/roboflow/models/inference.py
@@ -269,7 +269,7 @@ def predict_video(
             models.append(SUPPORTED_ADDITIONAL_MODELS[model])
 
         payload = json.dumps(
-            {"input_url": signed_url, "infer_fps": 5, "models": models}
+            {"input_url": signed_url, "infer_fps": fps, "models": models}
         )
 
         headers = {"Content-Type": "application/json"}
diff --git a/roboflow/models/video.py b/roboflow/models/video.py
@@ -135,7 +135,7 @@ def predict(
             models.append(SUPPORTED_ADDITIONAL_MODELS[model])
 
         payload = json.dumps(
-            {"input_url": signed_url, "infer_fps": 5, "models": models}
+            {"input_url": signed_url, "infer_fps": fps, "models": models}
         )
 
         response = requests.request("POST", url, headers=headers, data=payload)

Original file line number	Diff line number	Diff line change
`@@ -269,7 +269,7 @@ def predict_video(`
`269`	`269`	`models.append(SUPPORTED_ADDITIONAL_MODELS[model])`
`270`	`270`
`271`	`271`	`payload = json.dumps(`
`272`		`- {"input_url": signed_url, "infer_fps": 5, "models": models}`
	`272`	`+ {"input_url": signed_url, "infer_fps": fps, "models": models}`
`273`	`273`	`)`
`274`	`274`
`275`	`275`	`headers = {"Content-Type": "application/json"}`
Original file line number	Diff line number	Diff line change
`@@ -135,7 +135,7 @@ def predict(`
`135`	`135`	`models.append(SUPPORTED_ADDITIONAL_MODELS[model])`
`136`	`136`
`137`	`137`	`payload = json.dumps(`
`138`		`- {"input_url": signed_url, "infer_fps": 5, "models": models}`
	`138`	`+ {"input_url": signed_url, "infer_fps": fps, "models": models}`
`139`	`139`	`)`
`140`	`140`
`141`	`141`	`response = requests.request("POST", url, headers=headers, data=payload)`