Commit
·
1bfc058
1
Parent(s):
9ccf047
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ s_raw_model = YOLO("S-Raw.pt")
|
|
12 |
|
13 |
def snap(image, model, conf, iou):
|
14 |
# Convert the image to a numpy array
|
15 |
-
type(image)
|
16 |
image = np.array(image)
|
17 |
|
18 |
# Run the selected model
|
@@ -61,7 +61,7 @@ def snap(image, model, conf, iou):
|
|
61 |
|
62 |
demo = gr.Interface(
|
63 |
snap,
|
64 |
-
[gr.Image(source="webcam", tool=None, streaming=True), gr.
|
65 |
["image"],
|
66 |
title="Baybayin Instance Detection"
|
67 |
)
|
|
|
12 |
|
13 |
def snap(image, model, conf, iou):
|
14 |
# Convert the image to a numpy array
|
15 |
+
print(type(image))
|
16 |
image = np.array(image)
|
17 |
|
18 |
# Run the selected model
|
|
|
61 |
|
62 |
demo = gr.Interface(
|
63 |
snap,
|
64 |
+
[gr.Image(source="webcam", tool=None, streaming=True), gr.Radio(["M-Raw", "S-Raw", "N-Raw"]), gr.Slider(0, 1, value=0.6, label="Classifier Confidence Threshold"), gr.Slider(0, 1, value=0.7, label="IoU Threshold")],
|
65 |
["image"],
|
66 |
title="Baybayin Instance Detection"
|
67 |
)
|