Spaces:

jslin09
/

legal_document_drafting

Running on Zero

jslin09 commited on May 16, 2023

Commit

e4bbe61

1 Parent(s): 6fbb439

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from transformers import pipeline, set_seed
 import random
 model_name = "bloom-560m"
-model = AutoModelForCausalLM.from_pretrained(f"models/jslin09/{model_name}-finetuned-fraud")
 tokenizer = BloomTokenizerFast.from_pretrained(f'bigscience/{model_name}', bos_token = '<s>', eos_token = '</s>')
 def rnd_generate(prompt):

 import random
 model_name = "bloom-560m"
+model = AutoModelForCausalLM.from_pretrained(f'jslin09/{model_name}-finetuned-fraud')
 tokenizer = BloomTokenizerFast.from_pretrained(f'bigscience/{model_name}', bos_token = '<s>', eos_token = '</s>')
 def rnd_generate(prompt):