aryan365 commited on
Commit
aac5e8a
·
verified ·
1 Parent(s): 6abc47b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -43,8 +43,8 @@ def predict():
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
 
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