Update app.py

app.py

@@ -38,6 +38,38 @@ FTP_USER = "ford442"
 FTP_PASS = "GoogleBez12!"
 FTP_DIR = "1ink.us/stable_diff/"  # Remote directory on FTP server
 
+
+
+import subprocess
+
+def get_cuda_version():
+  """
+  Gets the CUDA version using the nvidia-smi command.
+
+  Returns:
+      str: The CUDA version as a string (e.g., "11.4"). 
+           Returns None if nvidia-smi is not found or fails.
+  """
+  try:
+    result = subprocess.run(['nvidia-smi'], capture_output=True, text=True)
+    output = result.stdout
+    # Find the line containing "CUDA Version"
+    for line in output.splitlines():
+      if "CUDA Version" in line:
+        # Extract the version number
+        return line.split(":")[-1].strip()
+    return None
+  except FileNotFoundError:
+    return None
+
+cuda_version = get_cuda_version()
+if cuda_version:
+  print(f"CUDA Version: {cuda_version}")
+else:
+  print("CUDA not found or nvidia-smi command failed.")
+
+
+    
 DESCRIPTIONXX = """
     ## ⚡⚡⚡⚡ REALVISXL V5.0 BF16 (Tester A) ⚡⚡⚡⚡
 """