Spaces:

psinha823
/

Major_project

Sleeping

psinha823 commited on Jun 13, 2024

Commit

3f70f9e

verified ·

1 Parent(s): 5e21dd3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -121,7 +121,7 @@ def classify_LC_image(img):
         # Making predictions
         model_h5 = tf.keras.models.load_model(model_path)
-        predictions = model_h5.predict(img_array, 0)
         # Getting the class with the highest probability
         class_idx = np.argmax(predictions)

         # Making predictions
         model_h5 = tf.keras.models.load_model(model_path)
+        predictions = model_h5.predict(img_array)
         # Getting the class with the highest probability
         class_idx = np.argmax(predictions)