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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import cv2
3
  import onnxruntime
4
  import numpy as np
5
 
6
- onnx_model_vgg19_path = "./vgg19.onnx"
7
  onnx_model_inceptionv3_path = "./InceptionV3-20epochs.onnx"
8
  onnx_model_resnet101_path = "./Resnet101-30epochs.onnx"
9
  onnx_model_vgg16_path = "./vgg16.onnx"
@@ -24,7 +24,7 @@ def predict_image(image_path, model):
24
  labels = class_labels
25
  model_resnet101 = onnxruntime.InferenceSession(onnx_model_resnet101_path)
26
  elif model == "Vgg19":
27
- img_size = (150, 150)
28
  labels = class_labels2
29
  model_vgg19 = onnxruntime.InferenceSession(onnx_model_vgg19_path)
30
 
 
3
  import onnxruntime
4
  import numpy as np
5
 
6
+ onnx_model_vgg19_path = "./vgg19-30epochs.onnx"
7
  onnx_model_inceptionv3_path = "./InceptionV3-20epochs.onnx"
8
  onnx_model_resnet101_path = "./Resnet101-30epochs.onnx"
9
  onnx_model_vgg16_path = "./vgg16.onnx"
 
24
  labels = class_labels
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