shawon100 commited on
Commit
ae44695
1 Parent(s): a58c48a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,5 +1,6 @@
1
  from transformers import pipeline
2
  import gradio as gr
 
3
  def generate_text(inp):
4
  output=pipe(inp, forced_bos_token_id=pipe.tokenizer.get_lang_id('bn'))
5
  tln=output[0]
1
  from transformers import pipeline
2
  import gradio as gr
3
+ pipe= pipeline('text2text-generation', model="facebook/m2m100_418M")
4
  def generate_text(inp):
5
  output=pipe(inp, forced_bos_token_id=pipe.tokenizer.get_lang_id('bn'))
6
  tln=output[0]