Update app.py

app.py

@@ -1,3 +1,5 @@
+import os
+os.system("nvidia-smi")
 # Step 2 - Importing Libraries
 from moviepy.editor import *
 from PIL import Image
@@ -23,10 +25,10 @@ title = "Video Story Generator with Audio by using dalle-mini and distilbart and
 tokenizer = AutoTokenizer.from_pretrained("sshleifer/distilbart-cnn-12-6")
 model = AutoModelForSeq2SeqLM.from_pretrained("sshleifer/distilbart-cnn-12-6")
 
-#device = "cuda:0" if torch.cuda.is_available() else "cpu"
-device = torch.device('cuda')
+device = "cuda:0" if torch.cuda.is_available() else "cpu"
+#device = torch.device('cuda')
 # transfer model
-model.to(device)
+#model.to(device)
 
 def get_output_video(text):
   inputs = tokenizer(text,