Update app.py

app.py

@@ -37,7 +37,7 @@ def generate_text(prompt, max_length, repetition_penalty, temperature):
     )
     
     # Decode the generated tokens and return the result
-    generated_text = tokenizer.batch_decode(outputs[:, input_ids.shape[[1]:-1], skip_special_tokens=True)[0].strip()
+    generated_text = tokenizer.batch_decode(outputs[:, input_ids.shape[1]:-1], skip_special_tokens=True)[0].strip()
     return generated_text
 
 # Create the Gradio interface