Update app.py

app.py

@@ -1,11 +1,11 @@
 import torch
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import streamlit as st
 
 # Load GPT-2 model and tokenizer
 model_name = "YasirAbdali/bart-summarization"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
 # Streamlit app
 st.title("Summary Generator")