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)
|