graceliying82 commited on
Commit
fe2ec56
·
1 Parent(s): c6b4b61

some adjustment

Browse files
app.py ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from object_detection import render_result_in_image
3
+
4
+ demo = gr.Interface(
5
+ fn=render_result_in_image,
6
+ inputs=gr.inputs.Image(type="pil", label="Upload Image"),
7
+ outputs=gr.outputs.Image(type="pil", label="Output Image with Predicted Instances")
8
+ )
9
+
10
+
11
+ if __name__ == "__main__":
12
+ demo.launch()
object-detection/object_detection.py → object_detection.py RENAMED
@@ -57,12 +57,3 @@ def render_result_in_image_helper(image, results, id2label):
57
  draw.text((box[0], box[1]), label_text, fill="white", font=font)
58
 
59
  return image
60
-
61
-
62
- demo = gr.Interface(
63
- fn=render_result_in_image,
64
- inputs=gr.inputs.Image(type="pil", label="Upload Image"),
65
- outputs=gr.outputs.Image(type="pil", label="Output Image with Predicted Instances")
66
- )
67
-
68
- demo.launch()
 
57
  draw.text((box[0], box[1]), label_text, fill="white", font=font)
58
 
59
  return image