Update app.py

app.py

@@ -8,6 +8,19 @@ from src.flux.generate import generate, seed_everything
 
 
 def process_image(api_key, seed, image, prompt):
+
+    import torch
+
+    # 检查是否有可用的 CUDA 设备
+    print("CUDA 是否可用:", torch.cuda.is_available())
+    
+    # 输出当前 CUDA 版本（如果可用）
+    if torch.cuda.is_available():
+        print("CUDA 版本:", torch.version.cuda)
+        print("当前 GPU 名称:", torch.cuda.get_device_name(0))  # 输出第 0 号 GPU 的名称
+        print("GPU 设备数量:", torch.cuda.device_count())
+    else:
+        print("PyTorch 未启用 CUDA，当前使用 CPU 运行。")
     if not api_key:
         raise gr.Error("❌ Please enter a valid OpenAI API key.")