jslin09 commited on
Commit
c203e02
1 Parent(s): 92f8d84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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
- #tokenizer = BloomTokenizerFast.from_pretrained(f'bigscience/{model_name}', bos_token = '<s>', eos_token = '</s>', pad_token = '<pad>')
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)