Update app.py

app.py

@@ -6,6 +6,7 @@ from requests.exceptions import RequestException
 import os
 import logging
 import json
+import GPUtil
 
 # 配置日志
 logging.basicConfig(level=logging.INFO)
@@ -14,6 +15,11 @@ logger = logging.getLogger(__name__)
 app = FastAPI()
 XAI_API_BASE = "https://api.x.ai/v1"
 
+cuda_available = True if len(GPUtil.getGPUs()) else False
+if cuda_available:
+    print(GPUtil.getGPUs()[0].name)
+print(cuda_available)
+
 class ChatRequest(BaseModel):
     messages: list[dict] = Field(..., description="消息列表")
     model: str = Field(..., description="模型ID")