Update app.py

app.py

@@ -160,7 +160,7 @@ def test():
     start_time = time.time()
     
     # model_save_directory = "./my_project_folder/pegasus_model"
-    model_save_directory = "./models/pegasus_model"
+    model_save_directory = "google/pegasus-xsum"
     
     model = PegasusForConditionalGeneration.from_pretrained(model_save_directory)
     tokenizer = PegasusTokenizer.from_pretrained(model_save_directory)
@@ -201,7 +201,7 @@ def bart():
     start_time = time.time()
     
     # Path to your BART model, adjust as necessary
-    model_save_directory = "./models/bart-large-cnn"
+    model_save_directory = "facebook/bart-large-cnn"
     
     # Load the tokenizer and model
     tokenizer = BartTokenizer.from_pretrained(model_save_directory)
@@ -265,7 +265,7 @@ def one():
     print(articles_list)
 
     # Path to your BART model
-    model_save_directory = "./models/bart-large-cnn"
+    model_save_directory = "facebook/bart-large-cnn"
     
     # Load the tokenizer and model
     tokenizer = BartTokenizer.from_pretrained(model_save_directory)