Update app.py

app.py

@@ -64,9 +64,9 @@ def get_output_video(text):
           is_reusable=True,
           is_verbose=True,
           dtype=torch.float16 if fp16 else torch.float32 ,#param ["float32", "float16", "bfloat16"] #float32 is faster than float16 but uses more GPU memory.
-          device='cuda' #'cpu' 
+          #device='cuda' #'cpu' 
       ) 
-      log_gpu_memory()
+      #log_gpu_memory()
 
       image = model.generate_image(
           text, 
@@ -85,7 +85,7 @@ def get_output_video(text):
       text=senten,
       seed=1,
       grid_size=1, #param {type:"integer"}
-      top_k=128, #param {type:"integer"}
+      top_k=256, #param {type:"integer"}
 
       image_path='generated',
       models_root='pretrained',