Update app.py

app.py

@@ -56,8 +56,9 @@ preds = model(input_text)
 
 # Select lab
 def get_label(prediction_tensor):
-    key = prediction_tensor.index(1)
-    return group_dict[key]
+    print(prediction_tensor.index(1))
+    #key = prediction_tensor.index(1)
+    #return group_dict[key]
     
 
 st.text(get_label(preds))