Update app.py

app.py

@@ -5,6 +5,7 @@ from huggingface_hub import login
 from fastapi import FastAPI, Request
 from fastapi.responses import JSONResponse
 import uvicorn
+import torch  # Thêm dòng import này
 
 # Cấu hình logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -23,7 +24,7 @@ try:
     qa_pipeline = pipeline(
         "question-answering",
         model="nguyenvulebinh/vi-mrc-base",
-        device=0 if torch.cuda.is_available() else -1
+        device=0 if torch.cuda.is_available() else -1  # Sử dụng torch đã import
     )
     logging.info("Model loaded successfully")
 except Exception as e:
@@ -52,4 +53,4 @@ async def api_answer(request: Request):
 
 if __name__ == "__main__":
     logging.info("Starting FastAPI...")
-    uvicorn.run(app, host="0.0.0.0", port=8000)  # Sử dụng cổng khác để test local, Spaces sẽ tự chọn cổng
+    uvicorn.run(app, host="0.0.0.0", port=8000)  # Sử dụng cổng khác để test