Update app.py

app.py

@@ -1,7 +1,9 @@
 import tensorflow as tf 
 from tensorflow.keras.models import Model
 from keras.models import load_model
-import gradio as gr 
+import gradio as gr
+from tensorflow.keras.preprocessing.image import load_img, img_to_array
+from tensorflow.keras.applications.densenet import preprocess_input, decode_predictions
 import numpy as np 
 model = load_model('Densenet.h5')
 model.load_weights("pretrained_model.h5")
@@ -18,13 +20,11 @@ def custom_decode_predictions(predictions, class_lables):
     return decoded_predictions
 
 def classify_image(img):
-    img_array = image.img_to_array(img)
-    img_array = np.expand_dims(img_array, axis=0)
+    img_array = img_to_array(img)
+    img_array = np.expand_dims(img_array, axis=0))
     img_array = preprocess_input(img_array)
-
-
-    predictions1 = model.predict(img_array)
-    decoded_predictions = custom_decode_predictions(predictions1, class_names)
+    predictions = model.predict(img_array)
+    decoded_predictions = custom_decode_predictions(predictions, class_names)
     return decoded_predictions
 
 # Gradio interface