Update app.py

app.py

@@ -1,5 +1,7 @@
 import streamlit as st
 from gnews import GNews
+from transformers import GPT2Tokenizer, GPT2LMHeadModel
+
 
 def fetch_news(topic):
     google_news = GNews(language='en', country='US')  # You can customize this
@@ -40,35 +42,43 @@ def page_trending_niche():
 
 
     
-from transformers import pipeline, set_seed
-generator = pipeline('text-generation', model='gpt2')
-set_seed(42)
+# Load pre-trained model and tokenizer
+tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
+model = GPT2LMHeadModel.from_pretrained('gpt2')
+
+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
+
 
 def page_social_media_generator():
-    st.title("Social Media Content Generator")
 
     # Using st.columns to create a two-column layout
     col1, col2 = st.columns([3, 1])
     with col1:
-        st.title("German Medical Content Manager")
+        st.title("Social Media Content Generator")
     with col2:
         st.image('Content_Creation_Pic.png', use_column_width=True)
-    
+
     # Retrieve the URL saved in the session state on the trending niche page
     article_url = st.session_state.get('selected_article_url', '')
-    
+
     if article_url:
         st.write(f"Selected Article URL: {article_url}")
-        input_text = st.text_area("Enter additional context or leave empty to use only the article URL:", value=article_url)
         
         if st.button('Generate Social Media Post'):
             with st.spinner('Generating...'):
-                post_content = generator(input_text, max_length=100, num_return_sequences=1)[0]['generated_text']
+                post_content = generate_social_media_post(article_url)
                 st.success('Generated Content:')
                 st.write(post_content)
     else:
         st.write("Please select an article from the 'What is trending in my niche?' page.")
 
+        
+
 
 def page_test():
     st.title('Test Page')