Update app.py

app.py

@@ -24,9 +24,9 @@ def generate_text(usertitle, content, max_length, temperature):
 #       'content': content
 #   }]
     input_text = f"[[[Title:]]]{usertitle}\n[[[Content:]]]{content}\n"
-    inputs = tokenizer(input_text, return_tensors='pt').to('cuda')    
-    generated_ids = model.generate(inputs['input_ids'], max_new_tokens=max_length, temperature=temperature, pad_token_id=tokenizer.eos_token_id)
-    generated_text = tokenizer.decode(generated_ids[0], skip_special_tokens=False)
+    inputs = tokenizer(input_text, return_tensors='pt').to('cuda')
+    generated_ids = model.generate(inputs['input_ids'], max_new_tokens=max_length, temperature=temperature)
+    generated_text = tokenizer.decode(generated_ids[0]).split(tokenizer.eos_token)[0]
     
     return generated_text