Update app.py

app.py

@@ -39,7 +39,7 @@ if user_input:
         output = model.generate(**inputs, max_length=200, do_sample=True, top_k=50, top_p=0.95)
     
     # Decode response
-    response = tokenizer.decode(output[:, inputs.input_ids.shape[-1]:][0], skip_special_tokens=True)
+    response = tokenizer.decode(output[:, inputs["input_ids"].shape[-1]:][0], skip_special_tokens=True)
     
     # Store and display chat history
     st.session_state.chat_history.append(("You", user_input))