Update app.py

app.py

@@ -4,10 +4,12 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 # Loading summarization model
 summarization_tokenizer = AutoTokenizer.from_pretrained("ieuniversity/sciencebrief_summarization")
 summarization_model = AutoModelForSeq2SeqLM.from_pretrained("ieuniversity/sciencebrief_summarization")
-summarizer = pipeline("summarization", tokenizer=summarization_tokenizer, model=summarization_model)
 
 def summarize(text):
-    return summarizer(text, max_length=120, min_length=30, do_sample=False)[0]['summary_text']
+    input_ids = summarization_tokenizer.encode(text, return_tensors="pt")
+    output_ids = summarization_model.generate(input_ids)
+    output_text = summarization_tokenizer.decode(output_ids[0], skip_special_tokens=True)
+    return output_text
 
 iface_summarize = gr.Interface(
     fn=summarize,