Spaces:
Sleeping
Sleeping
Fix gr.Inputs.Image type
Browse files
app.py
CHANGED
@@ -10,8 +10,7 @@ import gradio as gr
|
|
10 |
from ctrnet_infer import CTRNetInfer
|
11 |
|
12 |
|
13 |
-
def inference(
|
14 |
-
img_path = img.name
|
15 |
img = cv2.imread(img_path)
|
16 |
pred = ctrnet(img)
|
17 |
pred = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
|
@@ -29,10 +28,10 @@ examples = [['images/1.jpg'], ['images/2.jpg'], ['images/4.jpg']]
|
|
29 |
gr.Interface(
|
30 |
inference,
|
31 |
inputs=[
|
32 |
-
gr.inputs.Image(type='
|
33 |
],
|
34 |
outputs=[
|
35 |
-
gr.outputs.Image(type='
|
36 |
],
|
37 |
title=title,
|
38 |
description=description,
|
|
|
10 |
from ctrnet_infer import CTRNetInfer
|
11 |
|
12 |
|
13 |
+
def inference(img_path):
|
|
|
14 |
img = cv2.imread(img_path)
|
15 |
pred = ctrnet(img)
|
16 |
pred = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
|
|
|
28 |
gr.Interface(
|
29 |
inference,
|
30 |
inputs=[
|
31 |
+
gr.inputs.Image(type='filepath', label='Input'),
|
32 |
],
|
33 |
outputs=[
|
34 |
+
gr.outputs.Image(type='filepath', label='Output_image'),
|
35 |
],
|
36 |
title=title,
|
37 |
description=description,
|