Update app.py
Browse files
app.py
CHANGED
|
@@ -43,8 +43,8 @@ def predict():
|
|
| 43 |
yolo_predictions = model_yolo.predict(image) # YOLO prediction
|
| 44 |
|
| 45 |
return jsonify({
|
| 46 |
-
'
|
| 47 |
-
'
|
| 48 |
})
|
| 49 |
except Exception as e:
|
| 50 |
return jsonify({'error': str(e)}), 500
|
|
|
|
| 43 |
yolo_predictions = model_yolo.predict(image) # YOLO prediction
|
| 44 |
|
| 45 |
return jsonify({
|
| 46 |
+
'keras-prediction': predicted_class,
|
| 47 |
+
'yolo-prediction': yolo_predictions
|
| 48 |
})
|
| 49 |
except Exception as e:
|
| 50 |
return jsonify({'error': str(e)}), 500
|