Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,8 +6,7 @@ import random
|
|
| 6 |
|
| 7 |
model_name = "bloom-560m"
|
| 8 |
model = AutoModelForCausalLM.from_pretrained(f'jslin09/{model_name}-finetuned-fraud')
|
| 9 |
-
|
| 10 |
-
tokenizer = BloomTokenizerFast.from_pretrained(f'jslin09/{model_name}', bos_token = '<s>', eos_token = '</s>', pad_token = '<pad>')
|
| 11 |
|
| 12 |
def rnd_generate(prompt):
|
| 13 |
rnd_seed = random.randint(10, 500)
|
|
|
|
| 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 |
|
| 11 |
def rnd_generate(prompt):
|
| 12 |
rnd_seed = random.randint(10, 500)
|