Update pages/toxic.py

pages/toxic.py

@@ -1,15 +1,15 @@
 import streamlit as st
 import requests
 
-API_URL = "https://api-inference.huggingface.co/models/cointegrated/rubert-tiny-toxicity"
-headers = {"Authorization": f"Bearer {'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'}"}
+import streamlit as st
+import requests
+
+from transformers import pipeline
 
-def query(payload):
-    response = requests.post(API_URL, headers=headers, json=payload)
-    return response.json()
+pipe = pipeline("text-classification", model="cointegrated/rubert-tiny-toxicity")
 text = st.text_input("Введите комментарий")
 if text:
-    output = query(text)[0].get('label')
+    output = pipe(text)[0].get('label')
     if output == 'dangerous':
         st.markdown('<p style="color:red;">ОПАСНЫЙ КОММЕНТАРИЙ</p>', unsafe_allow_html=True)
     elif output == 'non-toxic':
@@ -19,4 +19,4 @@ if text:
     elif output == 'threat':
         st.markdown('<p style="color:red;">Угрожающий комментарий</p>', unsafe_allow_html=True)
     elif output == 'obscenity':
-        st.markdown('<p style="color:pink;">ууу, непристойности</p>', unsafe_allow_html=True)
+        st.markdown('<p style="color:pink;">ууу, непристойности</p>', unsafe_allow_html=True)