Update app.py

app.py

@@ -4,13 +4,20 @@ from transformers import pipeline
 
 
 
-pipe=pipeline('sentiment-analysis')
 
-def sentiment_calssifir(text):
+def translate(text,number):
+    if number==1:
+        model='Helsinki-NLP/opus-mt-en-es'
+    else if number==2:
+        model='Helsinki-NLP/opus-mt-en-fr'
+    else:
+        model='Helsinki-NLP/opus-mt-en-ru'
+
+    pipe=pipeline(model)  
     return  pipe(text)
 
 
 
 
-demo = gr.Interface(fn=sentiment_calssifir, inputs="text", outputs="json")
+demo = gr.Interface(fn=translate, inputs="text,number", outputs="json")
 demo.launch()