Update app.py

app.py

@@ -7,6 +7,19 @@ import os
 import torch
 import bitsandbytes
 
+# Function to wait for ZeroGPU to initialize
+def wait_for_gpu(timeout=60):
+    print("Waiting for ZeroGPU to initialize...")
+    start_time = time.time()
+    while not torch.cuda.is_available():
+        if time.time() - start_time > timeout:
+            raise RuntimeError("Timeout: ZeroGPU did not initialize within the expected time.")
+        time.sleep(2)  # Wait and check again
+    print("ZeroGPU is ready! Proceeding with model loading...")
+
+# Wait for GPU before loading the model
+wait_for_gpu()
+
 print(f"Is CUDA available: {torch.cuda.is_available()}")  # True
 print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")