Hugging Face's logo

Spaces:
onnx
/
EfficientNet-Lite4
Runtime error

App Files Community
2
akhaliq HF staff commited on
Commit
0922d82
·
1 Parent(s): c93fa57

Update app.py

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