Spaces:

Unearthed
/

Artifact_Classifier

Paused

Unearthed commited on Jan 17

Commit

e1855d3

verified ·

1 Parent(s): a7446ad

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ model = tf.keras.models.load_model("stone_artifact_with_nonartifact_classifier.h
 #model = tf.keras.models.load_model("stone_artifact_classifier.h5", compile=False)
 # Replace with your classes
-class_names = ["arrowheads", "axehammerstones", "scrapers", "notartifact"]
 IMG_SIZE = (224, 224)  # change this to your model's expected size

 #model = tf.keras.models.load_model("stone_artifact_classifier.h5", compile=False)
 # Replace with your classes
+class_names = ["arrowheads", "axehammerstones", "notartifact", "scrapers"]
 IMG_SIZE = (224, 224)  # change this to your model's expected size