swcrazyfan commited on
Commit
67a7715
·
1 Parent(s): 34c8686

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 T5Tokenizer, T5ForConditionalGeneration
5
  model = T5ForConditionalGeneration.from_pretrained('swcrazyfan/KingJamesify-T5-large')
6
  tokenizer = T5Tokenizer.from_pretrained('swcrazyfan/KingJamesify-T5-large')
7
 
8
- def king_jamesify(input_text, num_beams, max_length, temperature):
9
  input_ids = tokenizer.encode(input_text, return_tensors='pt').to(torch.int64)
10
  generated_ids = model.generate(input_ids=input_ids, max_length=max_length, num_beams=num_beams, temperature=temperature)
11
  result = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
 
5
  model = T5ForConditionalGeneration.from_pretrained('swcrazyfan/KingJamesify-T5-large')
6
  tokenizer = T5Tokenizer.from_pretrained('swcrazyfan/KingJamesify-T5-large')
7
 
8
+ def king_jamesify(input_text, temperature, max_length, num_beams):
9
  input_ids = tokenizer.encode(input_text, return_tensors='pt').to(torch.int64)
10
  generated_ids = model.generate(input_ids=input_ids, max_length=max_length, num_beams=num_beams, temperature=temperature)
11
  result = tokenizer.decode(generated_ids[0], skip_special_tokens=True)