Update app.py

app.py

@@ -1,5 +1,3 @@
-import os
-os.system("nvidia-smi")
 # Step 2 - Importing Libraries
 from moviepy.editor import *
 from PIL import Image
@@ -25,7 +23,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 = "cuda:0" if torch.cuda.is_available() else "cpu"
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+print(device)
+
 #device = torch.device('cuda')
 # transfer model
 #model.to(device)