Spaces:
Runtime error
Runtime error
ossaili
commited on
Commit
•
acbbc7d
1
Parent(s):
b96f427
na
Browse files
app.py
CHANGED
@@ -5,10 +5,6 @@ from models import efficientnetv2b0_25_arch_styles_Classifier
|
|
5 |
|
6 |
|
7 |
demo = gr.Interface(
|
8 |
-
<<<<<<< HEAD
|
9 |
efficientnetv2b0_25_arch_styles_Classifier, "image", "label")
|
10 |
-
=======
|
11 |
-
efficientnetv2b0_25_arch_styles_Classifier, "image", outputs=["text"])
|
12 |
-
>>>>>>> 6dfb3eb853ff6baf202a3117ebb5b61755454768
|
13 |
|
14 |
demo.launch()
|
|
|
5 |
|
6 |
|
7 |
demo = gr.Interface(
|
|
|
8 |
efficientnetv2b0_25_arch_styles_Classifier, "image", "label")
|
|
|
|
|
|
|
9 |
|
10 |
demo.launch()
|
models.py
CHANGED
@@ -41,16 +41,9 @@ def efficientnetv2b0_25_arch_styles_Classifier(image):
|
|
41 |
224, 224), interpolation=cv2.INTER_CUBIC)
|
42 |
img = tf.expand_dims(resized_image, 0)
|
43 |
efficientnetv2b0 = tf.keras.models.load_model(
|
44 |
-
<<<<<<< HEAD
|
45 |
"models\EfficientNetV2B0.h5")
|
46 |
|
47 |
y = efficientnetv2b0.predict(img).reshape(-1)
|
48 |
y = (np.round(y, 3)).tolist()
|
49 |
-
=======
|
50 |
-
"EfficientNetV2B0.h5")
|
51 |
-
|
52 |
-
y = efficientnetv2b0.predict(img).reshape(-1)
|
53 |
-
y = (np.round(y, 3)*100).tolist()
|
54 |
-
>>>>>>> 6dfb3eb853ff6baf202a3117ebb5b61755454768
|
55 |
|
56 |
return outputs(y)
|
|
|
41 |
224, 224), interpolation=cv2.INTER_CUBIC)
|
42 |
img = tf.expand_dims(resized_image, 0)
|
43 |
efficientnetv2b0 = tf.keras.models.load_model(
|
|
|
44 |
"models\EfficientNetV2B0.h5")
|
45 |
|
46 |
y = efficientnetv2b0.predict(img).reshape(-1)
|
47 |
y = (np.round(y, 3)).tolist()
|
|
|
|
|
|
|
|
|
|
|
|
|
48 |
|
49 |
return outputs(y)
|