Add GPU detection before running for logging

app.py

@@ -31,6 +31,20 @@ def read_logs():
     except Exception:
         return "ML worker not running"
 
+def detect_gpu():
+    try:
+        import torch
+        logger.info(f"PyTorch GPU: {torch.cuda.is_available()}")
+    except ImportError:
+        logger.warn("No PyTorch installed")
+    
+    try:
+        import tensorflow as tf
+        logger.info(f"Tensorflow GPU: {len(tf.config.list_physical_devices('GPU')) > 0}")
+    except ImportError:
+        logger.warn("No Tensorflow installed")
+
+detect_gpu()
 
 previous_url = ""
 ml_worker = None