Concepta commited on
Commit
08ca67b
1 Parent(s): e56db4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -5,6 +5,8 @@ from transformers import pipeline, set_seed
5
  import random
6
 
7
  model_name = "bloom-560m"
 
 
8
  model = AutoModelForCausalLM.from_pretrained(f'jslin09/{model_name}-finetuned-fraud')
9
  tokenizer = BloomTokenizerFast.from_pretrained(f'bigscience/{model_name}', bos_token = '<s>', eos_token = '</s>', pad_token = '<pad>')
10
 
 
5
  import random
6
 
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 = AutoModelForCausalLM.from_pretrained(f'jslin09/{model_name}-finetuned-fraud')
11
  tokenizer = BloomTokenizerFast.from_pretrained(f'bigscience/{model_name}', bos_token = '<s>', eos_token = '</s>', pad_token = '<pad>')
12