Update app.py

app.py

@@ -49,11 +49,9 @@ def model(img):
     predictions = loaded_model.predict(x)
     print(predictions)
 
-    predicted_class_index = 1 if (predictions[0]>0.5) else 0
-    print(predicted_class_index)
-
     predicted_class_index = np.argmax(predictions[0])
-
+    print(predicted_class_index)
+    
     return predicted_class_label
 
 cataract_app = gr.Interface(model,gr.Image(),["text"])