Update app.py

app.py

@@ -2,15 +2,10 @@ import gradio as gr
 from transformers import pipeline
 
 def do_action(text):
-    classifier = pipeline("zero-shot-classification",
-                          model="facebook/bart-large-mnli")
-
-    candidate_labels = ['gun control', 'abortion', 'gender transition', 'freedom of speech', 'immigration', 'taxes']
-    result = classifier(text, candidate_labels)
-
-    result = dict(zip(result['labels'], result['scores']))
+    pipe = pipeline("text-classification", model="papluca/xlm-roberta-base-language-detection")
+    result = pipe(text)
 
     return result
 
-iface = gr.Interface(fn=do_action, inputs="text", outputs="label")
+iface = gr.Interface(fn=do_action, inputs="text", outputs="text")
 iface.launch()