Spaces:

ossaili
/

architectural_styles

Runtime error

ossaili commited on Jul 1, 2022

Commit

e86e696

1 Parent(s): 4a35ff0

Update models.py

Files changed (1) hide show

models.py CHANGED Viewed

@@ -41,7 +41,7 @@ def efficientnetv2b0_25_arch_styles_Classifier(image):
         224, 224), interpolation=cv2.INTER_CUBIC)
     img = tf.expand_dims(resized_image, 0)
     efficientnetv2b0 = tf.keras.models.load_model(
-        "models\EfficientNetV2B0.h5")
     y = efficientnetv2b0.predict(img).reshape(-1)
     y = (np.round(y, 3)*100).tolist()

         224, 224), interpolation=cv2.INTER_CUBIC)
     img = tf.expand_dims(resized_image, 0)
     efficientnetv2b0 = tf.keras.models.load_model(
+        "EfficientNetV2B0.h5")
     y = efficientnetv2b0.predict(img).reshape(-1)
     y = (np.round(y, 3)*100).tolist()