AI-Assi / econ_analyzer
Cheongwoon's picture
Create econ_analyzer
d5c0829 verified
from flask import Flask, request, jsonify
from transformers import pipeline
app = Flask(__name__)
# βœ… 경제 해석 λͺ¨λΈ: 뢄석기 (analyzer)
econ_analyzer = pipeline(
"summarization", # 🀫 사싀은 μš”μ•½ νŒŒμ΄ν”„λΌμΈμ„ ν™œμš©ν•œ β€œν•΄μ„ 생성기” μ—­ν• 
model="digit82/kobart-summarization",
tokenizer="digit82/kobart-summarization"
)
@app.route("/")
def home():
return "πŸ“Š CW 경제 뢄석기 (KoBART 기반) λ™μž‘ 쀑"
@app.route("/econalyze", methods=["POST"])
def analyze_economy():
data = request.get_json()
econ_input = data.get("text", "")
if not econ_input.strip():
return jsonify({"error": "경제 μ§€ν‘œ μž…λ ₯이 μ—†μŠ΅λ‹ˆλ‹€"}), 400
try:
result = econ_analyzer(
econ_input,
max_length=300,
min_length=80,
do_sample=False
)
return jsonify({"analysis": result[0]["summary_text"]})
except Exception as e:
return jsonify({"error": str(e)}), 500
if __name__ == "__main__":
app.run(host="0.0.0.0", port=7861)