Update app.py

app.py

@@ -63,7 +63,7 @@ def chat(message: str, state: List[Dict[str, str]]) -> Generator[Tuple[List[Tupl
     else:
         # Generate tokens one by one
         with torch.no_grad():
-            for _ in range(50):  # Limit generation to 50 tokens
+            for _ in range(100):  # Limit generation to 50 tokens
                 outputs = model(input_ids)
                 next_token_logits = outputs.logits[:, -1, :]
                 next_token_id = torch.argmax(next_token_logits, dim=-1)