added json output for prediction

3 files changed, 18 insertions, 0 deletions

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..2b9922f
--- /dev/null
+++ b/main.py
@@ -0,0 +1,18 @@
+from roboflow import Roboflow
+import json
+rf = Roboflow(api_key="otKMUXAELXaU2XZsT77M")
+project = rf.workspace().project("biomedical-wastes")
+model = project.version(3).model
+
+# infer on a local image
+json_response = model.predict("./syringe.jpg", confidence=40, overlap=30).json()
+
+# save the JSON response to a file
+with open('output.json', 'w') as f:
+    json.dump(json_response, f)
+
+# visualize your prediction
+# model.predict("your_image.jpg", confidence=40, overlap=30).save("prediction.jpg")
+
+# infer on an image hosted elsewhere
+# print(model.predict("URL_OF_YOUR_IMAGE", hosted=True, confidence=40, overlap=30).json())
\ No newline at end of file
diff --git a/syringe.jpg b/syringe.jpg
new file mode 100644
index 0000000..83ddc67
--- /dev/null
+++ b/syringe.jpg
diff --git a/test.txt b/test.txt
deleted file mode 100644
index e69de29..0000000
--- a/test.txt
+++ /dev/null