dat257 commited on
Commit
c654a7e
·
verified ·
1 Parent(s): a7aa06f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -7,10 +7,10 @@ import torch
7
  from huggingface_hub import login
8
 
9
  # Cấu hình logging
10
- logging.basicConfig(level=logging.INFO)
11
 
12
  # Đăng nhập Hugging Face
13
- login(token="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") # Thay bằng token của bạn
14
 
15
  app = Flask(__name__)
16
 
@@ -28,10 +28,12 @@ def answer():
28
  data = request.json
29
  question = data.get("question")
30
  context = data.get("context")
 
31
  if not question or not context:
 
32
  return jsonify({"error": "Missing question or context"}), 400
33
  result = qa_pipeline(question=question, context=context)
34
- logging.info(f"Question: {question}, Answer: {result['answer']}")
35
  return jsonify({"answer": result["answer"]})
36
  except Exception as e:
37
  logging.error(f"API error: {e}")
 
7
  from huggingface_hub import login
8
 
9
  # Cấu hình logging
10
+ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
11
 
12
  # Đăng nhập Hugging Face
13
+ login(token=os.getenv("HF_TOKEN")) # Lấy token từ biến môi trường
14
 
15
  app = Flask(__name__)
16
 
 
28
  data = request.json
29
  question = data.get("question")
30
  context = data.get("context")
31
+ logging.info(f"Received request - Question: {question}, Context: {context[:200]}...")
32
  if not question or not context:
33
+ logging.error("Missing question or context")
34
  return jsonify({"error": "Missing question or context"}), 400
35
  result = qa_pipeline(question=question, context=context)
36
+ logging.info(f"Response - Answer: {result['answer']}")
37
  return jsonify({"answer": result["answer"]})
38
  except Exception as e:
39
  logging.error(f"API error: {e}")