Create roboflow_model.py

roboflow_model.py

@@ -0,0 +1,12 @@
+from inference import InferencePipeline
+
+# Initialize pipeline once
+pipeline = InferencePipeline.init_with_workflow(
+    api_key="dxkgGGHSZ3DI8XzVn29U",
+    workspace_name="naveen-kumar-hnmil",
+    workflow_id="detect-count-and-visualize-5"
+)
+
+def predict_image(image_path: str):
+    result = pipeline.infer(image_path)
+    return result["output_image"].numpy_image  # This is a NumPy image