Update app.py

app.py

@@ -1,9 +1,9 @@
 import streamlit as st
-from transformers import pipeline, TFAutoModelForSeq2SeqLM, T5Tokenizer
+from transformers import pipeline, AutoModelForSeq2SeqLM, T5Tokenizer
 
-# Load T5 model for summarization
+# Load T5 model for summarization using PyTorch
 model_name = "t5-small"
-model = TFAutoModelForSeq2SeqLM.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 tokenizer = T5Tokenizer.from_pretrained(model_name)
 summarizer = pipeline("summarization", model=model, tokenizer=tokenizer)