first commit

.gitignore

@@ -0,0 +1,3 @@
+flagged
+.idea
+venv

app.py

@@ -0,0 +1,14 @@
+import gradio as gr
+from transformers import pipeline
+
+reader = pipeline("text-classification", model='Guihss/TypeBERT', trust_remote_code=True,
+                  revision='main')
+
+
+
+def predict(text):
+    return {s['label']: s['score'] for s in reader(text, top_k=None)}
+
+
+iface = gr.Interface(fn=predict, inputs="text", outputs='label')
+iface.launch()

requirements.txt

@@ -0,0 +1,4 @@
+transformers
+torch
+torchvision
+torchaudio