Update app.py

app.py

@@ -28,10 +28,13 @@ import accelerate  # Add this for better memory management
 # =============================================
 # MEMORY OPTIMIZATION SETTINGS
 # =============================================
-# Enable memory efficient attention if available
-if hasattr(torch, 'backends') and hasattr(torch.backends, 'cuda') and torch.backends.cuda.is_enabled():
+# Check if CUDA is available and set memory optimization
+if torch.cuda.is_available():
+    print("✅ CUDA available, enabling GPU optimizations")
     torch.backends.cuda.enable_flash_sdp(True)
     torch.backends.cuda.enable_mem_efficient_sdp(True)
+else:
+    print("⚠️ CUDA not available, running on CPU")
 
 # Set environment variables for memory optimization
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:128"