Enutrof commited on
Commit
eac6f0a
·
1 Parent(s): 0421730

Updated app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def translate(model_name, source_sentence, num_beams):
40
  if isinstance(source_sentence, str):
41
  source_sentence = [source_sentence]
42
  model = models[model_name]
43
- predictions = model.predict(input, num_beams=int(num_beams))
44
  return [i.replace('▁', ' ') for i in predictions[0]]
45
 
46
  # Gradio interface
 
40
  if isinstance(source_sentence, str):
41
  source_sentence = [source_sentence]
42
  model = models[model_name]
43
+ predictions = model.predict(input)
44
  return [i.replace('▁', ' ') for i in predictions[0]]
45
 
46
  # Gradio interface