Spaces:
Runtime error
Runtime error
ossaili
commited on
Commit
·
efb8f14
1
Parent(s):
9358e26
na
Browse files
app.py
CHANGED
@@ -5,6 +5,6 @@ from models import efficientnetv2b0_25_arch_styles_Classifier
|
|
5 |
|
6 |
|
7 |
demo = gr.Interface(
|
8 |
-
efficientnetv2b0_25_arch_styles_Classifier, "image",
|
9 |
|
10 |
demo.launch()
|
|
|
5 |
|
6 |
|
7 |
demo = gr.Interface(
|
8 |
+
efficientnetv2b0_25_arch_styles_Classifier, "image", "label")
|
9 |
|
10 |
demo.launch()
|
models.py
CHANGED
@@ -44,6 +44,6 @@ def efficientnetv2b0_25_arch_styles_Classifier(image):
|
|
44 |
"models\EfficientNetV2B0.h5")
|
45 |
|
46 |
y = efficientnetv2b0.predict(img).reshape(-1)
|
47 |
-
y = (np.round(y, 3)
|
48 |
|
49 |
return outputs(y)
|
|
|
44 |
"models\EfficientNetV2B0.h5")
|
45 |
|
46 |
y = efficientnetv2b0.predict(img).reshape(-1)
|
47 |
+
y = (np.round(y, 3)).tolist()
|
48 |
|
49 |
return outputs(y)
|