Update app.py

app.py

@@ -78,16 +78,16 @@ model_type = st.radio(
 def highlight_obscene_words(text, model_type):
     if model_type=="FRIDA":
         result=model_wrapper(text, model_type)
-        st.markdown(f"{result}")
+        result=result.predictions
         for item in result:
-            if item[1][0].label=="non-toxic":
+            if item.label=="non-toxic":
                 st.markdown(
-                    "<span style='background:#47916B;'>{}:приемлемо</span>".format(item[1][0].text), 
+                    "<span style='background:#47916B;'>{}:приемлемо</span>".format(item.text), 
                     unsafe_allow_html=True
                 )
             else:
                 st.markdown(
-                    "<span style='background:#ffcccc;'>{}:токсично</span>".format(item[1][0].text), 
+                    "<span style='background:#ffcccc;'>{}:токсично</span>".format(item.text), 
                     unsafe_allow_html=True
                 )
     else: