Update app.py

app.py

@@ -18,7 +18,7 @@ from os import getcwd
 import glob
 import nltk
 nltk.download('punkt')
-
+device = "cuda:0" if torch.cuda.is_available() else "cpu"
 description = " Video Story Generator with Audio \n PS:  Generation of video by using Artifical Intellingence by dalle-mini and distilbart and gtss "
 title = "Video Story Generator with Audio by using dalle-mini and distilbart and gtss  "
 tokenizer = AutoTokenizer.from_pretrained("sshleifer/distilbart-cnn-12-6")
@@ -204,7 +204,7 @@ def get_output_video(text):
 text ='Once, there was a girl called Laura who went to the supermarket to buy the ingredients to make a cake. Because today is her birthday and her friends come to her house and help her to prepare the cake.'
 demo = gr.Blocks()
 with demo:
-    gr.Markdown("# Video Generator from long stories with Artificial Intelligence")
+    gr.Markdown("# Video Generator from stories with Artificial Intelligence")
     gr.Markdown("A story can be input by user. The story is summarized using DistillBART model. Then, then it is generated the images by using Dalle-mini and created the subtitles and audio gtts. These are generated as a video.")
     with gr.Row():
         # Left column (inputs)