Update src/inference.py

src/inference.py

@@ -91,10 +91,10 @@ class NewsInference:
                 item = {"value": value, "score": score}
                 data[label].append(item)
 
-        date = max(data["date"], key=lambda x: x['score'])
-        title = max(data["title"], key=lambda x: x['score'])
-        description = max(data["description"], key=lambda x: x['score'])
-        content = max(data["content"], key=lambda x: x['score'])
+        date = max(data["date"], key=lambda x: x['score']) if len(data["date"]) > 0 else None
+        title = max(data["title"], key=lambda x: x['score']) if len(data["title"]) > 0 else None
+        description = max(data["description"], key=lambda x: x['score']) if len(data["description"]) > 0 else None
+        content = max(data["content"], key=lambda x: x['score']) if len(data["content"]) > 0 else None
         response = {
             "date": date,
             "title": title,