Create app.py

app.py

@@ -0,0 +1,22 @@
+model_name = 'gngpostalsrvc/COHeN'
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+cls_explainer = SequenceClassificationExplainer(model, tokenizer)
+
+pattern = re.compile("[^\s\u05d0-\u05ea\u05b0-\u05bc\u05be\u05c1\u05c2\u05c7]")
+
+def predict(text):
+  text = " ".join([word for word in text.split() if word not in ['\u05e1', '\u05e4', '']])
+  text = re.sub(pattern, "", text)
+  text = sbl_normalization(text)
+  word_attributions = cls_explainer(text)
+  html = cls_explainer.visualize()
+  return html
+
+iface = gr.Interface(
+    fn=predict,
+    inputs='text',
+    outputs='html',
+)
+
+iface.launch()