Fix torch device for cpu

app.py

@@ -15,8 +15,9 @@ os.environ["PYTHONHASHSEED"] = "42"
 CUDA_AVAILABLE = torch.cuda.is_available()
 print(f"CUDA={CUDA_AVAILABLE}")
 device = "cuda" if CUDA_AVAILABLE else "cpu"
-print(f"count={torch.cuda.device_count()}")
-print(f"current={torch.cuda.get_device_name(torch.cuda.current_device())}")
+if CUDA_AVAILABLE:
+    print(f"count={torch.cuda.device_count()}")
+    print(f"current={torch.cuda.get_device_name(torch.cuda.current_device())}")
 
 continent_model = CLIPModel.from_pretrained(
     "jrheiner/thesis-clip-geoloc-continent",