aditi2222 commited on
Commit
33c23ee
·
1 Parent(s): fbf5bb0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers import AutoTokenizer, AutoModelWithLMHead, TranslationPipeline
7
 
8
  from transformers import pipeline
9
  import gradio as gr
10
- pipe= pipeline('text2text-generation', model="Helsinki-NLP/opus-mt-fr-en")
11
  def generate_text(inp):
12
  output=pipe(inp, forced_bos_token_id=pipe.tokenizer.get_lang_id('en'))
13
  tln=output[0]
 
7
 
8
  from transformers import pipeline
9
  import gradio as gr
10
+ pipe= pipeline('text2text-generation', model="SEBIS/legal_t5_small_trans_fr_en")
11
  def generate_text(inp):
12
  output=pipe(inp, forced_bos_token_id=pipe.tokenizer.get_lang_id('en'))
13
  tln=output[0]