datasciencedojo commited on
Commit
8f717b7
1 Parent(s): 5e6c8e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -8,7 +8,6 @@ torch_device = 'cuda' if torch.cuda.is_available() else 'cpu'
8
  tokenizer = PegasusTokenizer.from_pretrained(model_name)
9
  model = PegasusForConditionalGeneration.from_pretrained(model_name).to(torch_device)
10
 
11
-
12
  def get_response(input_text, num_return_sequences):
13
  batch = tokenizer.prepare_seq2seq_batch([input_text], truncation=True, padding='longest', max_length=10000,
14
  return_tensors="pt").to(torch_device)
 
8
  tokenizer = PegasusTokenizer.from_pretrained(model_name)
9
  model = PegasusForConditionalGeneration.from_pretrained(model_name).to(torch_device)
10
 
 
11
  def get_response(input_text, num_return_sequences):
12
  batch = tokenizer.prepare_seq2seq_batch([input_text], truncation=True, padding='longest', max_length=10000,
13
  return_tensors="pt").to(torch_device)