Update app.py

app.py

@@ -1,11 +1,19 @@
-import streamlit as st
-from transformers import pipeline
+# import streamlit as st
+# from transformers import pipeline
 
-st.title("Определятор токсичности!")
+import gradio as gr
 
-pipe = pipeline(task='text-classification', model='s-nlp/russian_toxicity_classifier')
-text = st.text_area('введите любой текст')
+def image_classifier(inp):
+    return {'cat': 0.3, 'dog': 0.7}
 
-if text:
-    out = pipe(text)
-    st.json(out)
+demo = gr.Interface(fn=image_classifier, inputs="image", outputs="label")
+demo.launch()
+
+# st.title("Определятор токсичности!")
+
+# pipe = pipeline(task='text-classification', model='s-nlp/russian_toxicity_classifier')
+# text = st.text_area('введите любой текст')
+
+# if text:
+#     out = pipe(text)
+#     st.json(out)