Update app.py

app.py

@@ -16,8 +16,15 @@ from huggingface_hub import hf_hub_download
 # بررسی دسترسی به GPU
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 dtype = torch.bfloat16 if torch.cuda.is_available() and torch.cuda.get_device_name(0).startswith("A100") else torch.float16
+
 if torch.cuda.is_available():
-    print(f"✅ CUDA is available! GPU: {torch.cuda.get_device_name(0)}")
+    num_gpus = torch.cuda.device_count()
+    if num_gpus > 1:
+        print(f"✅ {num_gpus} CUDA GPUs are available!")
+        for i in range(num_gpus):
+            print(f"GPU {i}: {torch.cuda.get_device_name(i)}")
+    else:
+        print(f"✅ CUDA is available! GPU: {torch.cuda.get_device_name(0)}")
 else:
     print("❌ No CUDA GPUs are available. Switching to CPU.")