Spaces:
Runtime error
Runtime error
yuragoithf
commited on
Commit
•
8a7486c
1
Parent(s):
0b3beca
Update app.py
Browse files
app.py
CHANGED
@@ -52,7 +52,7 @@ image_output = gr.outputs.Image(type="pil", label="Output Image")
|
|
52 |
# outputs = gr.outputs.HTML() #uncomment for single class output
|
53 |
|
54 |
|
55 |
-
def gen_pred(img
|
56 |
pil_image = img.convert('RGB')
|
57 |
open_cv_image = np.array(pil_image)
|
58 |
img = open_cv_image[:, :, ::-1].copy()
|
|
|
52 |
# outputs = gr.outputs.HTML() #uncomment for single class output
|
53 |
|
54 |
|
55 |
+
def gen_pred(img, model=seg_model):
|
56 |
pil_image = img.convert('RGB')
|
57 |
open_cv_image = np.array(pil_image)
|
58 |
img = open_cv_image[:, :, ::-1].copy()
|