karalif commited on
Commit
3c8522e
1 Parent(s): 2e7d843

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
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")