KilaruKusuma commited on
Commit
654cc18
·
verified ·
1 Parent(s): b0c89e5

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +29 -0
app.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from flask import Flask, request, jsonify
2
+ from transformers import pipeline
3
+
4
+ app = Flask(__name__)
5
+
6
+ # Load the summarization pipeline
7
+ pipe = pipeline("summarization", model="Falconsai/text_summarization")
8
+
9
+ @app.route('/')
10
+ def home():
11
+ return "Text Summarization API is running!"
12
+
13
+ @app.route('/summarize', methods=['POST'])
14
+ def summarize():
15
+ try:
16
+ data = request.json
17
+ text = data.get("text", "")
18
+
19
+ if not text:
20
+ return jsonify({"error": "No text provided"}), 400
21
+
22
+ summary = pipe(text, max_length=150, min_length=30, do_sample=False)
23
+ return jsonify({"summary": summary[0]['summary_text']})
24
+
25
+ except Exception as e:
26
+ return jsonify({"error": str(e)}), 500
27
+
28
+ if __name__ == '__main__':
29
+ app.run(debug=True)