Spaces:

veronhii
/

COS30082_Assignment

Build error

veronhii commited on Oct 20, 2023

Commit

912923f

1 Parent(s): 23ec869

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,8 @@ def predict_fn(img):
     x = np.expand_dims(x, axis=0)
     temp = model.predict(x)
-    return CLASS_LABEL[temp]
 model = tf.keras.models.load_model("model.h5")

     x = np.expand_dims(x, axis=0)
     temp = model.predict(x)
+    class_index = np.argmax(temp)
+    return CLASS_LABEL[class_index]
 model = tf.keras.models.load_model("model.h5")