Update app.py

add logging of predicted index

app.py

@@ -33,6 +33,7 @@ def decode_prediction(prediction):
         return "Error: Unexpected model output shape"
 
     predicted_index = np.argmax(prediction, axis=-1)[0]  # Get the index of the highest probability
+    print ("Predicted Index:", predicted_index)
     return CLASS_NAMES[predicted_index]  # Return the corresponding drug name
 
 # Function to preprocess the uploaded image
@@ -56,9 +57,6 @@ def predict_text(image):
     processed_image = preprocess_image(image)  # Ensure input is (64, 64, 3)
     prediction = model.predict(processed_image)
     
-    print("Model output shape:", prediction.shape)  # Should be (1, 78)
-    print("Model output values:", prediction)        # Print raw probabilities
-    
     # Decode the prediction to get the drug name
     predicted_text = decode_prediction(prediction)
     return predicted_text