Update app.py

app.py

@@ -8,7 +8,7 @@ import numpy as np
 from PIL import Image
 
 # Load the trained model
-model = tf.keras.models.load_model("prescription_model.h5")
+model = tf.keras.models.load_model("prescription_classification_model.keras")
 
 # Define the 78 drug names in the exact order corresponding to your model's output classes
 CLASS_NAMES = [