Hugging Face's logo

Spaces:
onnx
/
EfficientNet-Lite4
Runtime error

App Files Community
2
akhaliq HF staff commited on
Commit
276704f
·
1 Parent(s): 27b5604

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -56,7 +56,7 @@ def inference(img):
56
  img = cv2.imread(img)
57
  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
58
 
59
- img = pre_process_edgetpu(img, (128, 128, 3))
60
 
61
  img_batch = np.expand_dims(img, axis=0)
62
 
 
56
  img = cv2.imread(img)
57
  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
58
 
59
+ img = pre_process_edgetpu(img, (224, 224, 3))
60
 
61
  img_batch = np.expand_dims(img, axis=0)
62