Spaces:
Build error
Build error
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" | |
) | |
def home(): | |
return "π CW κ²½μ λΆμκΈ° (KoBART κΈ°λ°) λμ μ€" | |
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) |