davda54 commited on
Commit
62ba09f
·
1 Parent(s): 10dbfbb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -55,7 +55,7 @@ def translate(source, source_language, target_language):
55
  max_new_tokens = 512-1,
56
  do_sample=False
57
  )
58
- predictions = [tokenizer.decode(p, skip_special_tokens=True) for p in predictions.tolist()]
59
  return predictions[0]
60
 
61
 
@@ -101,8 +101,8 @@ with gr.Blocks(theme='sudeepshouche/minimalist') as demo:
101
  return {
102
  source: gr.update(interactive=True),
103
  submit: gr.update(interactive=True),
104
- source_language: gr.update(interactive=False),
105
- target_language: gr.update(interactive=False)
106
  }
107
 
108
 
 
55
  max_new_tokens = 512-1,
56
  do_sample=False
57
  )
58
+ predictions = [tokenizer.decode(p, skip_special_tokens=True).strip() for p in predictions.tolist()]
59
  return predictions[0]
60
 
61
 
 
101
  return {
102
  source: gr.update(interactive=True),
103
  submit: gr.update(interactive=True),
104
+ source_language: gr.update(interactive=True),
105
+ target_language: gr.update(interactive=True)
106
  }
107
 
108