Update app.py

app.py

@@ -1,15 +1,14 @@
 import os
-os.system('pip install streamlit transformers torch sentencepiece')
+os.system('pip install streamlit transformers torch')
 
 import streamlit as st
-from transformers import PegasusTokenizer, PegasusForConditionalGeneration
+from transformers import BartTokenizer, BartForConditionalGeneration
 
 # Load the model and tokenizer
-model_name =  '.'  # Path to the current directory where files are located
+model_name = 'llmahmad/facebook_BART_summary'
 
-
-tokenizer = PegasusTokenizer.from_pretrained(model_name)
-model = PegasusForConditionalGeneration.from_pretrained(model_name)
+tokenizer = BartTokenizer.from_pretrained(model_name)
+model = BartForConditionalGeneration.from_pretrained(model_name)
 
 def summarize_text(text):
     inputs = tokenizer(text, return_tensors="pt", truncation=True, padding="longest")