Update app.py

app.py

@@ -10,7 +10,10 @@ import hashlib
 import threading
 
 # Global settings
-SUMMARIZER_MODEL = "facebook/bart-large-cnn"  # Default summarization model
+SUMMARIZER_MODELS = {
+    "Default (facebook/bart-large-cnn)": "facebook/bart-large-cnn",
+    "Free Model (distilbart-cnn-6-6)": "sshleifer/distilbart-cnn-6-6"
+}
 CACHE_SIZE = 500  # Cache size for summaries
 RSS_FETCH_INTERVAL = timedelta(hours=8)
 ARTICLE_LIMIT = 5
@@ -30,7 +33,6 @@ NEWS_SOURCES = {
     }
 }
 
-# Initialize cache
 class NewsCache:
     def __init__(self, size):
         self.cache = {}
@@ -50,7 +52,7 @@ class NewsCache:
 
 cache = NewsCache(CACHE_SIZE)
 
-summarizer = pipeline("summarization", model=SUMMARIZER_MODEL, device=-1)
+summarizer = pipeline("summarization", model=SUMMARIZER_MODELS["Default (facebook/bart-large-cnn)"], device=-1)
 
 # Utility functions
 def fetch_rss_news(selected_sources):
@@ -162,3 +164,4 @@ if __name__ == "__main__":
     demo.launch()
 
 
+