nouman66 commited on
Commit
0fbc2ce
1 Parent(s): 4caf695

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -23,3 +23,11 @@ def translate_text(model, tokenizer, source_text, target_language):
23
  outputs = model.generate(**inputs, forced_bos_token_id=tokenizer.lang_code_to_id[target_language])
24
  translated_text = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
25
  return translated_text
 
 
 
 
 
 
 
 
 
23
  outputs = model.generate(**inputs, forced_bos_token_id=tokenizer.lang_code_to_id[target_language])
24
  translated_text = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
25
  return translated_text
26
+ from transformers import AutoConfig, AutoModelForSeq2SeqLM, AutoTokenizer
27
+
28
+ model_name = "facebook/mbart-large-50"
29
+ AutoConfig.from_pretrained(model_name).clear_cache()
30
+ AutoModelForSeq2SeqLM.from_pretrained(model_name).clear_cache()
31
+ AutoTokenizer.from_pretrained(model_name).clear_cache()
32
+ pip install sentencepiece
33
+