Update app.py

app.py

@@ -39,14 +39,17 @@ def predict_image(image_path, model):
     img = np.expand_dims(img, axis=0)  # Add batch dimension
     img = img / 255.0  # Normalize pixel values
     if model == "Vgg19":
-        prediction = model_vgg19.run(None, {"input": img})[0]
+        input_name = model_vgg19.get_inputs()[0].name
+        output_name= model_vgg19.get_outputs()[0].name
+        prediction = model_vgg19.run(None, {input_name: img})
+        
     elif model == "Vgg16":
         prediction = model_vgg16.run(None, {"input": img})[0]
     elif model == "InceptionV3":
         prediction = model_inceptionv3.run(None, {"input": img})[0]
     elif model == "Resnet101":
         prediction = model_resnet101.run(None, {"input": img})[0]
-    prediction = np.argmax(prediction)
+    prediction = np.argmax(prediction[0][0])
     pre_class = labels[prediction]
 
     return pre_class