Update models.py

summarizer/models.py

@@ -1,15 +1,7 @@
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 def load_model():
-    model_name = "m3hrdadfi/bert2bert-fa-summarizer"
-
+    model_name = "nafisehNik/mt5-persian-summary"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-
-    summarizer = pipeline("summarization", model=model, tokenizer=tokenizer)
-
-    def summarize(text):
-        result = summarizer(text, max_length=128, min_length=30, do_sample=False)
-        return result[0]["summary_text"]
-
-    return summarize
+    return tokenizer, model