diff --git a/src/__init__.py b/src/__init__.py index 7ef36a1..36850dc 100644 --- a/src/__init__.py +++ b/src/__init__.py @@ -40,7 +40,7 @@ def worker_loop(hostname, model): # Using the result of the prediction assing a classification # TODO - In the future these possibilities should be fetchable from the data service - if round(pred) == 1: + if round(pred) == 0: response_dict['classification'] = "covid-19" else: response_dict['classification'] = "not covid-19"