Update app.py

app.py

@@ -8,33 +8,28 @@ model = AutoModelForCausalLM.from_pretrained(model_name)
 
 # Function to generate tweets
 def generate_tweet(input_text):
-    prompt = ("You are a tech-savvy, forward-thinking individual with a deep understanding of technology, innovation, and cultural trends. "
-              "Craft a tweet that reflects insightful commentary, wit, or actionable advice based on the following idea: \"{}\". "
-              "Ensure the response is concise, engaging, and suitable for a diverse audience on social media. "
-              "Incorporate elements of thought leadership, futuristic perspectives, and practical wisdom where appropriate.").format(input_text)
-
-    inputs = tokenizer(prompt, return_tensors="pt", max_length=512, truncation=True, padding=True)
+    inputs = tokenizer(input_text, return_tensors="pt", max_length=512, truncation=True, padding=True)
     outputs = model.generate(
         inputs['input_ids'],
         attention_mask=inputs['attention_mask'],
-        max_length=280,
+        max_length=150,  # Limit to 150 tokens for brevity
         num_return_sequences=1,
-        top_p=0.95,
-        top_k=50,
+        top_p=0.9,  # Narrow focus to ensure more concise results
+        top_k=40,   # Focus on high-probability words
         do_sample=True,
         pad_token_id=tokenizer.pad_token_id
     )
     generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
     # Extract the tweet text (exclude prompt if included)
-    return generated_text.replace(prompt, "").strip()
+    return generated_text.strip()
 
 # Gradio interface
 def main():
     with gr.Blocks() as interface:
         gr.Markdown("""
         # Tweet Generator
-        Enter a topic or idea, and the AI will craft a tweet inspired by innovative, philosophical, and tech-savvy thought leadership.
+        Enter a topic or idea, and the AI will craft a concise, engaging, and impactful tweet inspired by innovative thought leadership.
         """)
 
         with gr.Row():
@@ -50,4 +45,4 @@ def main():
 # Run Gradio app
 if __name__ == "__main__":
     app = main()
-    app.launch(share=True)
+    app.launch(share=True)