diff --git a/src/app.py b/src/app.py index b921d1c..93639a8 100644 --- a/src/app.py +++ b/src/app.py @@ -40,7 +40,7 @@ async def on_message(message: IncomingMessage): r.raise_for_status() fields = { - k: float(v) for k, v in msg_json['all_predictions'] + k: float(v) for k, v in msg_json['all_predictions'].items() } fields['_classification_duration'] = float(msg_json['classification_duration'])