Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import base64
 import json
 import ast
@@ -6,6 +7,11 @@ from io import BytesIO
 import types
 import sys
 
+# Force CPU-only & disable bitsandbytes CUDA checks in this environment
+os.environ.setdefault("CUDA_VISIBLE_DEVICES", "")
+os.environ.setdefault("BITSANDBYTES_NOWELCOME", "1")
+os.environ.setdefault("BITSANDBYTES_DISABLE_CUDA_CHECK", "1")
+
 import torch
 import torchvision.transforms as T
 from PIL import Image