Update app.py

app.py

@@ -1,19 +1,11 @@
-import gradio as gr
-from gradio.mix import Series
+from transformers import RobertaTokenizer, RobertaForSequenceClassification
 
-description = "Reguetton spanish songs toxicity classification"
-title = "ES-EN Translation / Toxicity classification / EN-ES Translation"
+# load tokenizer and model weights
+tokenizer = RobertaTokenizer.from_pretrained('SkolkovoInstitute/roberta_toxicity_classifier')
+model = RobertaForSequenceClassification.from_pretrained('SkolkovoInstitute/roberta_toxicity_classifier')
 
-translator_es = gr.Interface.load("huggingface/Helsinki-NLP/opus-mt-es-en")
-story_gen = gr.Interface.load("huggingface/SkolkovoInstitute/roberta_toxicity_classifier")   
-translator_en = gr.Interface.load("huggingface/Helsinki-NLP/opus-mt-en-es")
+# prepare the input
+batch = tokenizer.encode('you are amazing', return_tensors='pt')
 
-examples = [["La aventura comienza en"]]
-
-interface = Series(translator_es, story_gen, translator_en, description = description,
-      title = title,
-      examples=examples, 
-      inputs = gr.inputs.Textbox(lines = 10)
-)
-
-interface.launch()
+# inference
+model(batch)