updated output labels

app.py

@@ -20,7 +20,8 @@ A worker was using a grinder in a confined space when he became dizzy from the f
 
 def predict(txt):
     pred,_,probs = learn.predict(txt)
-    return {labels[i]: float(probs[i]) for i in range(1)}
+    return {labels[i]: float(probs[i]) for i in range(len(labels))}
+    # return {labels[i]: float(probs[i]) for i in range(1)}
     #return learn.predict(txt)
     #  formating  of return f"Rating: {pred} prob: float(probs[pred_idx[i]]"
     #return f'{pred}'