Update app.py(1)

app.py

@@ -1,17 +1,16 @@
 from flask import Flask, request, jsonify
-from transformers import pipeline
+from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
 import os, time
 
 app = Flask(__name__)
 
-# Ensure model is cached in persistent directory
-os.environ["TRANSFORMERS_CACHE"] = "/data/models"
+# Load from local 'model/' folder
+tokenizer = AutoTokenizer.from_pretrained("./model")
+model = AutoModelForSeq2SeqLM.from_pretrained("./model")
+summarizer = pipeline("summarization", model=model, tokenizer=tokenizer)
 
-# Load model
-summarizer = pipeline("summarization", model="Falconsai/text_summarization")
-
-# Warm-up
-summarizer("This is a warm-up request.", max_length=30, min_length=5, do_sample=False)
+# Warm up
+summarizer("Warm-up input.", max_length=30, min_length=5, do_sample=False)
 
 @app.route("/summarize", methods=["POST"])
 def summarize():