Cheongwoon commited on
Commit
d377ca8
ยท
verified ยท
1 Parent(s): 7ed09e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -8,7 +8,7 @@ summarizer = pipeline("summarization", model="digit82/kobart-summarization", tok
8
 
9
  @app.route("/")
10
  def home():
11
- return "CW KoBART ์š”์•ฝ ์„œ๋ฒ„ ์ž‘๋™ ์ค‘ (๋ฉ€ํ‹ฐ ์—”๋“œํฌ์ธํŠธ /summarize, /econalyze)"
12
 
13
  @app.route("/summarize", methods=["POST"])
14
  def summarize():
@@ -34,5 +34,16 @@ def econalyze():
34
  except Exception as e:
35
  return jsonify({"error": str(e)}), 500
36
 
 
 
 
 
 
 
 
 
 
 
 
37
  if __name__ == "__main__":
38
  app.run(host="0.0.0.0", port=7860)
 
8
 
9
  @app.route("/")
10
  def home():
11
+ return "CW KoBART ์š”์•ฝ ์„œ๋ฒ„ ์ž‘๋™ ์ค‘ (๋ฉ€ํ‹ฐ ์—”๋“œํฌ์ธํŠธ /summarize, /econalyze, /ideas)"
12
 
13
  @app.route("/summarize", methods=["POST"])
14
  def summarize():
 
34
  except Exception as e:
35
  return jsonify({"error": str(e)}), 500
36
 
37
+ @app.route("/ideas", methods=["POST"])
38
+ def ideas():
39
+ data = request.get_json()
40
+ keywords = data.get("keywords", [])
41
+ if not keywords:
42
+ return jsonify({"ideas": []})
43
+ try:
44
+ return jsonify({"ideas": [f"{kw} ๊ด€๋ จ ๋ธ”๋กœ๊ทธ ์•„์ด๋””์–ด" for kw in keywords]})
45
+ except Exception as e:
46
+ return jsonify({"error": str(e)}), 500
47
+
48
  if __name__ == "__main__":
49
  app.run(host="0.0.0.0", port=7860)