Update app.py
Browse files
app.py
CHANGED
@@ -1,2 +1,9 @@
|
|
1 |
import pkg_resources
|
2 |
-
from transformers import pipeline
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
import pkg_resources
|
2 |
+
from transformers import pipeline
|
3 |
+
import gradio as gr
|
4 |
+
|
5 |
+
translator = pipeline("translation", model="Helsinki-NLP/opus-mt-is-en")
|
6 |
+
sentiment_classifier = pipeline("text-classification", model="Birkir/electra-base-igc-is-sentiment-analysis")
|
7 |
+
formality_classifier = pipeline("text-classification", model="svanhvit/formality-classification-icebert")
|
8 |
+
detoxify_pipeline = pipeline('text-classification', model='unitary/toxic-bert', tokenizer='bert-base-uncased', function_to_apply='sigmoid', return_all_scores=True)
|
9 |
+
politeness_classifier = pipeline("text-classification", model="Genius1237/xlm-roberta-large-tydip")
|