remove inputs_ids in tokenizer.decode

app.py

@@ -28,7 +28,7 @@ def generate_text(input_text,max_new_tokens=512,top_k=50,top_p=0.95,temperature=
             output = model.generate(input_ids, attention_mask=attention_mask, max_new_tokens=max_new_tokens, top_k=top_k, top_p=top_p, temperature=temperature,do_sample=True)
     else:
         output = model.generate(input_ids, attention_mask=attention_mask, max_new_tokens=max_new_tokens, top_k=top_k, top_p=top_p, temperature=temperature,do_sample=True)
-    return tokenizer.decode(output[0][input_ids['input_ids'].shape[1]:], skip_special_tokens=True)
+    return tokenizer.decode(output[0], skip_special_tokens=True)
 
 
 time_story = 0