swcrazyfan commited on
Commit
5c2ce00
·
1 Parent(s): 7528b8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -7,6 +7,7 @@ tokenizer = T5Tokenizer.from_pretrained('swcrazyfan/KingJamesify-T5-large')
7
 
8
  def king_jamesify(input_text, num_beams, max_length, temperature):
9
  prompt_text = "kingify: " + input_text
 
10
  input_ids = tokenizer.encode(prompt_text, return_tensors='pt').to(torch.int64)
11
  generated_ids = model.generate(input_ids=input_ids, max_length=max_length, num_beams=num_beams, num_return_sequences=num_beams, temperature=temperature)
12
  result = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
 
7
 
8
  def king_jamesify(input_text, num_beams, max_length, temperature):
9
  prompt_text = "kingify: " + input_text
10
+ numrs = num_beams - 1
11
  input_ids = tokenizer.encode(prompt_text, return_tensors='pt').to(torch.int64)
12
  generated_ids = model.generate(input_ids=input_ids, max_length=max_length, num_beams=num_beams, num_return_sequences=num_beams, temperature=temperature)
13
  result = tokenizer.decode(generated_ids[0], skip_special_tokens=True)