Update app.py

app.py

@@ -1,7 +1,14 @@
-import torch
 from transformers import AutoImageProcessor, AutoModelForObjectDetection, pipeline
 from PIL import Image, ImageDraw
 import gradio as gr
+import torch
+
+if torch.cuda.is_available():
+    # Only initialize GPU-specific components if a GPU is available
+    from your_library import ZeroGPU
+    ZeroGPU.initialize()
+else:
+    print("GPU not available, running on CPU. Skipping ZeroGPU initialization.")
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print(f"Using device: {device}")