Update app.py

app.py

@@ -21,13 +21,16 @@ with zipfile.ZipFile(zip_model_path, 'r') as zip_ref:
 # After unzipping, the model should be in a specific directory, check the directory structure
 model_path = os.path.join(model_dir, 'Bart_model')
 
+# Print out the model_path for debugging
+print("Model Path:", model_path)
+
 # Verify that the directory exists and contains the necessary files
 if not os.path.exists(model_path):
     st.error(f"Model directory {model_path} does not exist or is incorrect.")
 else:
     # Load the tokenizer and model
-    tokenizer = BartTokenizer.from_pretrained(model_name)
-    model = BartForConditionalGeneration.from_pretrained(model_name)
+    tokenizer = BartTokenizer.from_pretrained('facebook/bart-large-cnn')
+    model = BartForConditionalGeneration.from_pretrained(model_path)
 
     # Create a summarization pipeline
     summarizer = pipeline("summarization", model=model, tokenizer=tokenizer)
@@ -50,4 +53,4 @@ else:
             st.subheader("Generated Summary")
             st.write(generated_summary)
         else:
-            st.warning("Please enter some text to generate a summary.")
+            st.warning("Please enter some text to generate a summary.")