Spaces:

AfshinMA
/

Image_Classifier_Keras

Sleeping

AfshinMA commited on Sep 17, 2024

Commit

9947da8

verified ·

1 Parent(s): 44cc280

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ def classify_image(input_image):
         labels = ["Unknown"] * 1000  # Fallback in case the request fails
     # Load the MobileNetV2 model
-    inception_net = keras.applications.MobileNetV2(
         input_shape=(224, 224, 3),
         alpha=1.0,
         include_top=True,

         labels = ["Unknown"] * 1000  # Fallback in case the request fails
     # Load the MobileNetV2 model
+    inception_net = tf.keras.applications.MobileNetV2(
         input_shape=(224, 224, 3),
         alpha=1.0,
         include_top=True,