Update app.py

app.py

@@ -37,7 +37,7 @@ def generate_answer():
       inputs = tokenizer(st.session_state.input_text+ tokenizer.eos_token, return_tensors="pt")
       result = model.generate(**inputs, max_length=1000)
       message_bot = tokenizer.decode(
-          result[:, inputs.shape[-1]:][0], skip_special_tokens=True
+          result[0], skip_special_tokens=True
       )
       
     st.session_state.history.append({"message": user_message, "is_user": True})