Update app.py

app.py

@@ -751,6 +751,9 @@ model_zero_init = False
 
 @spaces.GPU()
 def apply_nf4_storage():
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    torch_device = torch.device(device)
+    
     global model, model_zero_init
     if not model_zero_init:
         model = model.to(torch_device)