Update app.py

app.py

@@ -377,8 +377,15 @@ def signal_handler(signum, frame):
     sys.exit(0)
 
 if __name__ == "__main__":
-    # Initialize ZeroGPU
-    initialize_zero_gpu()
+   def initialize_zero_gpu():
+    """Initialize Hugging Face ZeroGPU (if available)"""
+    if torch.cuda.is_available():
+        # Install CUDA drivers and adjust library versions if necessary
+        # ... (refer to solutions above)
+        deepspeed.init_distributed()
+        print("ZeroGPU initialized successfully.")
+    else:
+        print("CUDA is not available. Using CPU instead.")
 
     # Register cleanup handlers
     atexit.register(cleanup)