Update app.py

app.py

@@ -4,10 +4,10 @@ import time
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 # Streamlit page configuration
-st.set_page_config(page_title="Review Keypoint Extractor", page_icon="🔑")
+st.set_page_config(page_title="Review Keypoint Extractor (BART-Large-CNN)", page_icon="🔑")
 
 # Define the model
-model_name = "t5-small"
+model_name = "facebook/bart-large-cnn"
 
 # Cache the model and tokenizer to avoid reloading
 @st.cache_resource
@@ -24,8 +24,8 @@ def generate_keypoint(review, max_new_tokens=64):
     
     start_time = time.time()
     
-    # T5-specific prompt
-    prompt = f"summarize: {review}"
+    # BART-specific prompt (no additional prompt engineering)
+    prompt = review
     
     # Inference
     inputs = tokenizer(prompt, return_tensors="pt", truncation=True, padding=True).to(device)
@@ -41,8 +41,8 @@ def generate_keypoint(review, max_new_tokens=64):
     return keypoint, elapsed
 
 # Streamlit UI
-st.title("🔑 Review Keypoint Extractor")
-st.write("Enter a product review below to extract its key points using the T5-Small model.")
+st.title("🔑 Review Keypoint Extractor (BART-Large-CNN)")
+st.write("Enter a product review below to extract its key points using the facebook/bart-large-cnn model.")
 
 # Input field for review
 review = st.text_area("Product Review", placeholder="e.g., The Jackery power station is lightweight and charges quickly, but the battery life could be longer.")