Update app.py

app.py

@@ -1,15 +1,14 @@
 import os
+os.system('pip install streamlit transformers torch sentencepiece')
+
 import streamlit as st
 from transformers import PegasusTokenizer, PegasusForConditionalGeneration
 
-# Install necessary packages (only needed if not already installed)
-os.system('pip install streamlit transformers torch')
-
 # Load the model and tokenizer
-model_path = '.'  # Path to the current directory where files are located
+model_name = "google/pegasus-xsum"  # Using a different Pegasus model
 
-tokenizer = PegasusTokenizer.from_pretrained(model_path)
-model = PegasusForConditionalGeneration.from_pretrained(model_path)
+tokenizer = PegasusTokenizer.from_pretrained(model_name)
+model = PegasusForConditionalGeneration.from_pretrained(model_name)
 
 def summarize_text(text):
     inputs = tokenizer(text, return_tensors="pt", truncation=True, padding="longest")
@@ -37,7 +36,7 @@ if __name__ == "__main__":
         <style>
         .reportview-container {
             flex-direction: row;
-            justify-content: center;
+            justify-content: center.
         }
         </style>
         """,