Concepta commited on
Commit
3846d7f
1 Parent(s): d20c3c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import random
7
  #model_name = "bloom-560m"
8
  #for RAG: stjiris/bert-large-portuguese-cased-legal-mlm-nli-sts-v1
9
  #for tts: rhaymison/transcription-portuguese-legal-whisper-peft
10
- model_name = MBZUAI/LaMini-T5-738M
11
  model = AutoModelForSeq2SeqLM.from_pretrained(f'{model_name}')
12
  #tokenizer = BloomTokenizerFast.from_pretrained(f'{model_name}', bos_token = '<s>', eos_token = '</s>', pad_token = '<pad>')
13
  tokenizer = AutoTokenizer.from_pretrained(f'{model_name}')
 
7
  #model_name = "bloom-560m"
8
  #for RAG: stjiris/bert-large-portuguese-cased-legal-mlm-nli-sts-v1
9
  #for tts: rhaymison/transcription-portuguese-legal-whisper-peft
10
+ model_name = "MBZUAI/LaMini-T5-738M"
11
  model = AutoModelForSeq2SeqLM.from_pretrained(f'{model_name}')
12
  #tokenizer = BloomTokenizerFast.from_pretrained(f'{model_name}', bos_token = '<s>', eos_token = '</s>', pad_token = '<pad>')
13
  tokenizer = AutoTokenizer.from_pretrained(f'{model_name}')