Updated app.py

app.py

@@ -36,11 +36,11 @@ models = {
     "BBGM Model": bbgm_model
 }
 
-def translate(model_name, source_sentence, num_beams):
+def translate(model_name, source_sentence):
     if isinstance(source_sentence, str):
         source_sentence = [source_sentence]
     model = models[model_name]
-    predictions = model.predict(input)
+    predictions = model.predict(source_sentence)
     return [i.replace('▁', ' ') for i in predictions[0]]
 
 # Gradio interface
@@ -49,7 +49,6 @@ interface = gr.Interface(
     inputs=[
         gr.Dropdown(choices=["BM Model", "BBGM Model"], label="Model Selection"),
         gr.Textbox(placeholder="Enter English sentence here...", label="Source Sentence"),
-        gr.Slider(minimum=1, maximum=10, default=5, step=1, label="Number of Beams"),
     ],
     outputs=[
         gr.Textbox(label="Prediction 1"),
@@ -60,13 +59,15 @@ interface = gr.Interface(
     description='Type your English text in the left text box to get 🇳🇬 Pidgin translations on the right. '
                 'Tell us the best translation by clicking one of the buttons below.',
     examples=[
-        ['BBGM Model', 'Who are you?', 1],
-        ['BBGM Model', 'Is a personal philosophy of moral relativism, the only way to survive in this ethically complex world, or is it just an excuse to justify doing bad things?', 1],
-        ['BBGM Model', 'Is a personal philosophy of moral relativism, the only way to survive in this ethically complex world, or is it just an excuse to justify doing bad things?', 5],
-        ['BBGM Model', 'I know every song by that artiste.', 1],
-        ['BBGM Model', 'They should not be permitted here.', 1],
-        ['BBGM Model', 'I am lost, please help me find my way to the market.', 1],
-        ['BM Model', 'I am lost, please help me find my way to the market.', 1],
+        ['BBGM Model', 'Who are you?'],
+        ['BBGM Model', 'Is a personal philosophy of moral relativism, the only way to survive in this ethically complex world, or is it just an excuse to justify doing bad things?'],
+        ['BM Model', 'Is a personal philosophy of moral relativism, the only way to survive in this ethically complex world, or is it just an excuse to justify doing bad things?'],
+        ['BBGM Model', 'I know every song by that artiste.'],
+        ['BM Model', 'I know every song by that artiste.'],
+        ['BBGM Model', 'They should not be permitted here.'],
+        ['BM Model', 'They should not be permitted here.'],
+        ['BBGM Model', 'I am lost, please help me find my way to the market.'],
+        ['BM Model', 'I am lost, please help me find my way to the market.'],
     ]
 )