Update app.py

app.py

@@ -41,22 +41,20 @@ def page_trending_niche():
 
 
 
-    
-# Load pre-trained model and tokenizer
-tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
-model = GPT2LMHeadModel.from_pretrained('gpt2')
+from transformers import pipeline
 
-def generate_social_media_post(article_url):
-    prompt = f"Write a social media post about this article: {article_url}"
-    inputs = tokenizer.encode(prompt, return_tensors='pt')
-    outputs = model.generate(inputs, max_length=100, num_return_sequences=1, no_repeat_ngram_size=2)
-    post = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return post
+# Initialize the summarization pipeline with BART
+summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+
+def generate_social_media_post(article_text):
+    # Ensure the article text is not too long for the model
+    article_text = article_text[:1024]  # BART has a limit on the length of the text it can process
+    summary = summarizer(article_text, max_length=130, min_length=30, do_sample=False)[0]['summary_text']
+    return summary
 
 
-def page_social_media_generator():
 
-    # Using st.columns to create a two-column layout
+def page_social_media_generator():
     col1, col2 = st.columns([3, 1])
     with col1:
         st.title("Social Media Content Generator")
@@ -69,9 +67,13 @@ def page_social_media_generator():
     if article_url:
         st.write(f"Selected Article URL: {article_url}")
         
+        # Example placeholder for article text fetching
+        # In a real application, replace the following line with code to fetch the article's text from `article_url`
+        article_text = "The full text of the article goes here."
+        
         if st.button('Generate Social Media Post'):
             with st.spinner('Generating...'):
-                post_content = generate_social_media_post(article_url)
+                post_content = generate_social_media_post(article_text)
                 st.success('Generated Content:')
                 st.write(post_content)
     else: