Update app.py

app.py

@@ -1,24 +1,44 @@
-import gradio as gr
+# import gradio as gr
 
-from transformers import pipeline
+# from transformers import pipeline
+
+
+
+
+# def translate(text,number):
+#     model1=''
+#     if number==1:
+#         model1='Helsinki-NLP/opus-mt-en-es'
+#     elif number==2:
+#         model1='Helsinki-NLP/opus-mt-en-fr'
+#     else:
+#         model1='Helsinki-NLP/opus-mt-en-ru'
+
+#     pipe=pipeline("translation", model=model1)  
+#     return  pipe(text)[0]["translation_text"]
 
 
 
 
-def translate(text,number):
-    model1=''
-    if number==1:
-        model1='Helsinki-NLP/opus-mt-en-es'
-    elif number==2:
-        model1='Helsinki-NLP/opus-mt-en-fr'
-    else:
-        model1='Helsinki-NLP/opus-mt-en-ru'
+# demo = gr.Interface(fn=translate, inputs=["text","number"], outputs="json")
+# demo.launch()
 
-    pipe=pipeline("translation", model=model1)  
-    return  pipe(text)[0]["translation_text"]
 
 
 
+import gradio as gr
+
+from transformers import pipeline
+
+pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")
+
+def predict(text):
+  return pipe(text)[0]["translation_text"]
+
+demo = gr.Interface(
+  fn=predict, 
+  inputs='text',
+  outputs='text',
+)
 
-demo = gr.Interface(fn=translate, inputs=["text","number"], outputs="json")
 demo.launch()