Update app.py

app.py

@@ -9,7 +9,8 @@ def round_to_2(num):
     return round(num, 2)
 
 def format_output(out_list):
-    white = 0
+    if len(out_list) == 1:
+        out_list = out_list[0]
     for dictionary in out_list:
         if dictionary["label"] =='W':
             white = round_to_2(dictionary["score"])