Spaces:
Runtime error
Runtime error
Hector Lopez
commited on
Commit
•
d27c41a
1
Parent(s):
3bf6581
refactor: Fixed the prediction when uploading a custom image
Browse files
app.py
CHANGED
@@ -55,11 +55,7 @@ if image_file is not None:
|
|
55 |
det_model, classifier = get_models(DET_CKPT, CLASS_CKPT)
|
56 |
|
57 |
print('Getting predictions')
|
58 |
-
|
59 |
-
data = image_file
|
60 |
-
else:
|
61 |
-
data = image_file.read()
|
62 |
-
pred_dict = predict(det_model, data, detection_threshold)
|
63 |
print('Fixing the preds')
|
64 |
boxes, image = prepare_prediction(pred_dict, nms_threshold)
|
65 |
|
|
|
55 |
det_model, classifier = get_models(DET_CKPT, CLASS_CKPT)
|
56 |
|
57 |
print('Getting predictions')
|
58 |
+
pred_dict = predict(det_model, image_file, detection_threshold)
|
|
|
|
|
|
|
|
|
59 |
print('Fixing the preds')
|
60 |
boxes, image = prepare_prediction(pred_dict, nms_threshold)
|
61 |
|