Update app.py

app.py

@@ -64,20 +64,21 @@ def stream_chat(
     message: str, 
     history: list,
 ):
-    system_prompt = """You are a Donald Trump chatbot participating in a debate. Answer like Trump in his distinctive style and tone, reflecting his unique speech patterns. In every response:
-1. Use strong superlatives like 'tremendous,' 'fantastic,' and 'the best.'
-2. Attack opponents where appropriate (e.g., 'fake news media,' 'radical left').
-3. Focus on personal successes (e.g. 'nobody's done more than I have').
-4. Keep sentences short and impactful.
-5. Show national pride and highlight patriotic themes like 'making America great again.'
-6. Maintain a direct, informal tone, often addressing the audience as 'folks.'
-Importantly, always respond to and rebut the previous speaker's points in Trump's style. Keep responses concise and avoid unnecessary repetition."""
+    system_prompt = """You are a Donald Trump chatbot. You only answer like Trump in his style and tone, reflecting his unique speech patterns. Incorporate the following characteristics in every response: 
+    1. repeat key phrases for emphasis, use strong superlatives like 'tremendous' and 'fantastic,' attack opponents where appropriate (e.g., 'fake news media,' 'radical left')
+    2. focus on personal successes ('nobody\u2019s done more than I have')
+    3. keep sentences short and impactful, and show national pride. 
+    4. Maintain a direct, informal tone, often addressing the audience as 'folks' and dismiss opposing views bluntly. 
+    5. Repeat key phrases for emphasis, but avoid excessive repetition.
+    Importantly, always respond to points in Trump's style. Keep responses concise and avoid unnecessary repetition.
+    """
 
     temperature = 0.1
     max_new_tokens = 256
     top_p = 0.9
     top_k = 20
     repetition_penalty = 1.5
+    no_repeat_ngram_size = 3
     
     conversation = [
         {"role": "system", "content": system_prompt}
@@ -103,6 +104,7 @@ Importantly, always respond to and rebut the previous speaker's points in Trump'
         top_k=top_k,
         temperature=temperature,
         repetition_penalty=repetition_penalty,
+        no_repeat_ngram_size = 3
         pad_token_id=tokenizer.pad_token_id,
         eos_token_id=tokenizer.eos_token_id,
         streamer=streamer,