Subhajit Chakraborty commited on
Commit
37a43bb
·
1 Parent(s): 7edfda4

added files

Browse files
Files changed (2) hide show
  1. app.py +19 -0
  2. requirements.txt +3 -0
app.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from flask import Flask, request, jsonify
2
+ from transformers import pipeline
3
+
4
+ app = Flask(__name__)
5
+
6
+ summarizer = pipeline("summarization", model="Falconsai/text_summarization")
7
+
8
+ @app.route("/summarize", methods=["POST"])
9
+ def summarize():
10
+ try:
11
+ data = request.get_json()
12
+ text = data["text"]
13
+ summary = summarizer(text, max_length=150, min_length=30, do_sample=False)
14
+ return jsonify({"summary": summary[0]["summary_text"]})
15
+ except Exception as e:
16
+ return jsonify({"error": str(e)}), 400
17
+
18
+ if __name__ == "__main__":
19
+ app.run(host="0.0.0.0", port=7860)
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ flask
2
+ transformers
3
+ torch