Create inference.py

inference.py

@@ -0,0 +1,18 @@
+from roboflow import Roboflow
+from PIL import Image
+import numpy as np
+import json
+
+# Load Roboflow config
+with open("roboflow_config.json", "r") as f:
+    config = json.load(f)
+
+# Initialize Roboflow model
+rf = Roboflow(api_key=config["api_key"])
+project = rf.workspace(config["workspace"]).project(config["project"])
+model = project.version(config["version"]).model
+
+def predict(image):
+    image = np.array(image)
+    prediction = model.predict(image).plot()
+    return Image.fromarray(prediction)