Update app.py

app.py

@@ -8,16 +8,16 @@ model = AutoModelForCausalLM.from_pretrained("Manasa1/gpt-finetuned-tweets")
 tokenizer = AutoTokenizer.from_pretrained("Manasa1/gpt-finetuned-tweets")
 
 
-# Define the function to generate tweets
 def generate_tweet():
-    prompt = "Generate a tweet that reflects the personality in the fine-tuned dataset:"
-    inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
+    prompt = "Write a concise, creative tweet reflecting the style and personality in the fine-tuned dataset."
+    inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=100)
     outputs = model.generate(
         inputs["input_ids"],
-        max_length=280,
+        max_length=140,
         num_return_sequences=1,
-        top_p=0.9,
-        temperature=0.7
+        top_p=0.8,
+        temperature=0.6,
+        repetition_penalty=1.2,  # Penalizes repetitive tokens
     )
     generated_tweet = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return generated_tweet.strip()