Update app.py

app.py

@@ -23,6 +23,7 @@ description = " Video Story Generator with Audio \n PS:  Generation of video by
 title = "Video Story Generator with Audio by using dalle-mini and distilbart and gtss  "
 tokenizer = AutoTokenizer.from_pretrained("sshleifer/distilbart-cnn-12-6")
 model = AutoModelForSeq2SeqLM.from_pretrained("sshleifer/distilbart-cnn-12-6")
+model     = model.to(device)
 def get_output_video(text):
   inputs = tokenizer(text, 
                     max_length=1024, 
@@ -32,7 +33,7 @@ def get_output_video(text):
   summary_ids = model.generate(inputs["input_ids"])
   summary = tokenizer.batch_decode(summary_ids, 
                                   skip_special_tokens=True, 
-                                  clean_up_tokenization_spaces=False)
+                                  clean_up_tokenization_spaces=False).to(device)
   plot = list(summary[0].split('.'))
 
   def generate_image(