app.py

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)