Update app.py

app.py

@@ -32,13 +32,13 @@ def textMode(text, count):
     prediction = tokenizer.batch_decode(
     phi2_text.generate(
         **inputs, 
-        max_new_tokens=500,
+        max_new_tokens=count,
         bos_token_id=tokenizer.bos_token_id, 
         eos_token_id=tokenizer.eos_token_id,
         pad_token_id=tokenizer.pad_token_id
     )
     )
-    return prediction[0].rstrip('<|endoftext|>').rstrip("\n")[:count]
+    return prediction[0].rstrip('<|endoftext|>').rstrip("\n")