ossaili commited on
Commit
efb8f14
1 Parent(s): 9358e26
Files changed (2) hide show
  1. app.py +1 -1
  2. models.py +1 -1
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", outputs=["text"])
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)*100).tolist()
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)