Update app.py

app.py

@@ -44,11 +44,17 @@ def predict_image(image_path, model):
         prediction = model_vgg19.run(None, {input_name: img})
         
     elif model == "Vgg16":
-        prediction = model_vgg16.run(None, {"input": img})[0]
+        input_name = model_vgg16.get_inputs()[0].name
+        output_name= model_vgg16.get_outputs()[0].name
+        prediction = model_vgg16.run(None, {input_name: img})
     elif model == "InceptionV3":
-        prediction = model_inceptionv3.run(None, {"input": img})[0]
+        input_name = model_inceptionv3.get_inputs()[0].name
+        output_name= model_inceptionv3.get_outputs()[0].name
+        prediction = model_inceptionv3.run(None, {input_name: img})
     elif model == "Resnet101":
-        prediction = model_resnet101.run(None, {"input": img})[0]
+        input_name = model_resnet101.get_inputs()[0].name
+        output_name= model_resnet101.get_outputs()[0].name
+        prediction = model_resnet101.run(None, {input_name: img})
     prediction = np.argmax(prediction[0][0])
     pre_class = labels[prediction]