Update app.py

app.py

@@ -5,6 +5,9 @@ import modin.pandas as pd
 from PIL import Image
 from diffusers import DiffusionPipeline 
 
+
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
+
 if torch.cuda.is_available():
     PYTORCH_CUDA_ALLOC_CONF={'max_split_size_mb': 6000}
     torch.cuda.max_memory_allocated(device=device)