Update app.py

app.py

@@ -1,17 +1,17 @@
+import gradio as gr
+
 from transformers import pipeline
-fill_mask = pipeline(
-    "fill-mask",
-    model="alger-ia/dziribert",
-    tokenizer="alger-ia/dziribert"
+
+pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")
+
+def predict(text):
+  return pipe(text)[0]["translation_text"]
+
+iface = gr.Interface(
+  fn=predict, 
+  inputs='text',
+  outputs='text',
+  examples=[["Hello! My name is Omar"]]
 )
-generated_text_set = set()
-text= "هذا [MASK]"
-for i in range(50):
-    result = fill_mask(text)
-    r = result[random.randint(0, 4)]["sequence"]
-    if r not in generated_text_set:
-        print(r)
-        text = r + " [MASK]"
-        generated_text_set.add(r)
-    else:
-        continue
+
+iface.launch()