Ruslan Magana Vsevolodovna commited on
Commit
9e7dce3
·
1 Parent(s): 2efab13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def get_output_video(text):
33
  summary_ids = model.generate(inputs["input_ids"])
34
  summary = tokenizer.batch_decode(summary_ids,
35
  skip_special_tokens=True,
36
- clean_up_tokenization_spaces=False).to(device)
37
  plot = list(summary[0].split('.'))
38
 
39
  def generate_image(
 
33
  summary_ids = model.generate(inputs["input_ids"])
34
  summary = tokenizer.batch_decode(summary_ids,
35
  skip_special_tokens=True,
36
+ clean_up_tokenization_spaces=False)
37
  plot = list(summary[0].split('.'))
38
 
39
  def generate_image(