Update app.py

app.py

@@ -24,9 +24,8 @@ model = AutoModelForSeq2SeqLM.from_pretrained("sshleifer/distilbart-cnn-12-6")
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 print(device)
 
-def log_gpu_memory():
-    print(subprocess.check_output('nvidia-smi').decode('utf-8'))
-
+#def log_gpu_memory():
+#    print(subprocess.check_output('nvidia-smi').decode('utf-8'))
 #log_gpu_memory()
 
 
@@ -65,7 +64,7 @@ def get_output_video(text):
           models_root=models_root,
           is_reusable=True,
           is_verbose=True,
-          dtype=torch.float16 if fp16 else torch.float32 ,#param ["float32", "float16", "bfloat16"] 
+          dtype=torch.float16 if fp16 else torch.float32, #param ["float32", "float16", "bfloat16"] 
           #device='cuda' #'cpu' 
       ) 
       #log_gpu_memory()