Update app.py

app.py

@@ -2,11 +2,10 @@ from flask import Flask, request, jsonify
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 import os
 
-# --- Khởi tạo Flask ---
 app = Flask(__name__)
 
-# --- Load mô hình tóm tắt tiếng Việt ---
-model_name = "VietAI/vit5-large-vietnews-summarization"  # dùng dấu '-' chuẩn
+# Dùng mô hình nhỏ hơn để tiết kiệm bộ nhớ
+model_name = "VietAI/vit5-base-vietnews-summarization"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
@@ -16,7 +15,6 @@ summarizer = pipeline(
     tokenizer=tokenizer
 )
 
-# --- Route home ---
 @app.route("/")
 def home():
     return jsonify({
@@ -24,7 +22,6 @@ def home():
         "usage": "POST /summarize với JSON: { 'text': '...', 'topic': '...' }"
     })
 
-# --- Route tóm tắt ---
 @app.route("/summarize", methods=["POST"])
 def summarize():
     data = request.get_json(force=True)
@@ -33,6 +30,11 @@ def summarize():
         return jsonify({"error": "Thiếu trường 'text'"}), 400
 
     text = data["text"]
+    topic = data.get("topic", "").strip()  # Nếu không có topic, mặc định trống
+
+    # Nếu có topic, ghép vào đầu văn bản để mô hình hiểu hướng tóm tắt
+    if topic:
+        text = f"Chủ đề: {topic}. Nội dung: {text}"
 
     # Giới hạn số từ để tránh lỗi bộ nhớ
     max_words = 1000
@@ -49,7 +51,6 @@ def summarize():
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-# --- Chạy app ---
 if __name__ == "__main__":
     port = int(os.environ.get("PORT", 7860))
     app.run(host="0.0.0.0", port=port)