Spaces:
Sleeping
Sleeping
File size: 986 Bytes
9fd7d89 3b4df89 7507a36 524f780 9fd7d89 17c487a 9fd7d89 e116825 3b4df89 9fd7d89 3b4df89 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
import os
from flask import Flask, request, jsonify
from transformers import pipeline
# Ensure HF doesn't request a token
os.environ["HF_HOME"] = "/app/cache"
os.environ["HF_HUB_DISABLE_SYMLINKS_WARNING"] = "1"
os.environ["HF_HUB_DISABLE_TELEMETRY"] = "1"
os.environ["HF_HUB_OFFLINE"] = "0"
# Load model
summarizer = pipeline("summarization", model="t5-base")
app = Flask(__name__)
@app.route("/")
def home():
return "Summarization API is running!"
@app.route("/summarize", methods=["POST"])
def summarize_text():
data = request.get_json()
text = data.get("text", "")
max_length = data.get("max_length", 50)
min_length = data.get("min_length", 10)
if not text:
return jsonify({"error": "No text provided"}), 400
summary = summarizer(text, max_length=max_length, min_length=min_length, do_sample=False)
return jsonify(summary)
if __name__ == "__main__":
print("🚀 API is running on port 7860")
app.run(host="0.0.0.0", port=7860)
|