Spaces:

Supisss
/

Docker-ASHOKA

Runtime error

Supisss commited on Jan 22

Commit

3dee800

1 Parent(s): bd20b63

revisi error model 1.2

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ def build_model():
     base_model = MobileNetV2(
         input_shape=(224, 224, 3),
         include_top=False,
-        weights=None
     )
     base_model.trainable = False

     base_model = MobileNetV2(
         input_shape=(224, 224, 3),
         include_top=False,
+        weights="imagenet"
     )
     base_model.trainable = False