Cheongwoon commited on
Commit
b1359ca
Β·
verified Β·
1 Parent(s): d5c0829

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +27 -13
app.py CHANGED
@@ -1,25 +1,39 @@
1
  from flask import Flask, request, jsonify
2
  from transformers import pipeline
3
 
4
- app = Flask(__name__)
 
 
5
 
6
  summarizer = pipeline("summarization", model="digit82/kobart-summarization", tokenizer="digit82/kobart-summarization")
7
 
8
  @app.route("/")
9
  def home():
10
- return "βœ… CW KoBART ON"
11
 
12
  @app.route("/summarize", methods=["POST"])
13
  def summarize():
14
- data = request.get_json()
15
- text = data.get("text", "")
16
- if not text.strip():
17
- return jsonify({"error": "No Text"}), 400
18
- try:
19
- result = summarizer(text, max_length=200, min_length=40, do_sample=False)
20
- return jsonify({"summary": result[0]["summary_text"]})
21
- except Exception as e:
22
- return jsonify({"error": str(e)}), 500
 
 
 
 
 
 
 
 
 
 
 
 
23
 
24
- if __name__ == "__main__":
25
- app.run(host="0.0.0.0", port=7860)
 
1
  from flask import Flask, request, jsonify
2
  from transformers import pipeline
3
 
4
+ app = Flask(name)
5
+
6
+ βœ… KoBART μš”μ•½κΈ° 곡톡 μ‚¬μš©
7
 
8
  summarizer = pipeline("summarization", model="digit82/kobart-summarization", tokenizer="digit82/kobart-summarization")
9
 
10
  @app.route("/")
11
  def home():
12
+ return "βœ… CW KoBART μš”μ•½ μ„œλ²„ μž‘λ™ 쀑 (λ©€ν‹° μ—”λ“œν¬μΈνŠΈ /summarize, /econalyze)"
13
 
14
  @app.route("/summarize", methods=["POST"])
15
  def summarize():
16
+ data = request.get_json()
17
+ text = data.get("text", "")
18
+ if not text.strip():
19
+ return jsonify({"error": "μš”μ•½ν•  ν…μŠ€νŠΈκ°€ μ—†μŠ΅λ‹ˆλ‹€"}), 400
20
+ try:
21
+ result = summarizer(text, max_length=200, min_length=40, do_sample=False)
22
+ return jsonify({"summary": result[0]["summary_text"]})
23
+ except Exception as e:
24
+ return jsonify({"error": str(e)}), 500
25
+
26
+ @app.route("/econalyze", methods=["POST"])
27
+ def analyze_economy():
28
+ data = request.get_json()
29
+ econ_input = data.get("text", "")
30
+ if not econ_input.strip():
31
+ return jsonify({"error": "경제 μ§€ν‘œ μž…λ ₯이 μ—†μŠ΅λ‹ˆλ‹€"}), 400
32
+ try:
33
+ result = summarizer(econ_input, max_length=300, min_length=80, do_sample=False)
34
+ return jsonify({"analysis": result[0]["summary_text"]})
35
+ except Exception as e:
36
+ return jsonify({"error": str(e)}), 500
37
 
38
+ if name == "main":
39
+ app.run(host="0.0.0.0", port=7860)