Anas090 commited on
Commit
be22087
·
verified ·
1 Parent(s): 382ef4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def predict_image(image_path, model):
25
  model_resnet101 = onnxruntime.InferenceSession(onnx_model_resnet101_path)
26
  elif model == "Vgg19":
27
  img_size = (250, 200)
28
- labels = class_labels2
29
  model_vgg19 = onnxruntime.InferenceSession(onnx_model_vgg19_path)
30
 
31
  elif model == "Vgg16":
 
25
  model_resnet101 = onnxruntime.InferenceSession(onnx_model_resnet101_path)
26
  elif model == "Vgg19":
27
  img_size = (250, 200)
28
+ labels = class_labels
29
  model_vgg19 = onnxruntime.InferenceSession(onnx_model_vgg19_path)
30
 
31
  elif model == "Vgg16":