feat: initial version

app.py

@@ -1,7 +1,21 @@
 import gradio as gr
+import torch
+from transformers import pipeline
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+model = pipeline('onnx', model='model/best.onnx')
+
+def object_detection(image):
+    results = model(image)
+
+    return {
+        'image': image,
+        'detections': results
+    }
+
+inputs = gr.inputs.Image()
+outputs = gr.outputs.Image(label='Input Image', type='numpy'), gr.outputs.JSON(label='Object Detections')
+
+iface = gr.Interface(fn=object_detection, inputs=inputs, outputs=outputs)
+
 iface.launch()

model/best.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bde60c3139326613178f27bcdc47baf8124380722e9328987bd26390c30262a4
+size 28552986

requirements.txt

@@ -0,0 +1,2 @@
+torch==2.0.0
+git+https://github.com/huggingface/transformers.git