Update app.py

app.py

@@ -4,11 +4,13 @@ from transformers import BloomTokenizerFast
 from transformers import pipeline, set_seed
 import random
 
-model_name = "bloom-560m"
+#model_name = "bloom-560m"
 #for RAG: stjiris/bert-large-portuguese-cased-legal-mlm-nli-sts-v1
 #for tts: rhaymison/transcription-portuguese-legal-whisper-peft
-model = AutoModelForCausalLM.from_pretrained(f'jslin09/{model_name}-finetuned-fraud')
-tokenizer = BloomTokenizerFast.from_pretrained(f'bigscience/{model_name}', bos_token = '<s>', eos_token = '</s>', pad_token = '<pad>')
+model_name = MBZUAI/LaMini-T5-738M
+model = AutoModelForSeq2SeqLM.from_pretrained(f'{model_name}')
+#tokenizer = BloomTokenizerFast.from_pretrained(f'{model_name}', bos_token = '<s>', eos_token = '</s>', pad_token = '<pad>')
+tokenizer = AutoTokenizer.from_pretrained(f'{model_name}')
 
 def rnd_generate(prompt):
     rnd_seed = random.randint(10, 500)