jslin09 commited on
Commit
e4bbe61
1 Parent(s): 6fbb439

Update app.py

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