Update app.py

app.py

@@ -106,7 +106,7 @@ if st.session_state["qa_pipeline"]:
             with st.spinner("Generating response..."):
                 try:
                     # Generate the model response for general QA (T5 model)
-                    response = st.session_state["qa_pipeline"](f"Q: {user_input}", max_length=400)
+                    response = st.session_state["qa_pipeline"](f"Q: {user_input}", max_length=250)
                     generated_text = response[0]["generated_text"]
 
                     # Update the conversation
@@ -128,7 +128,7 @@ else:
             value=st.session_state["user_input"],
             key="calculation_input",
         )
-        if st.button("Send Calculation", key="send_calculation_button"):
+        if st.button("Submit", key="send_calculation_button"):
             if user_input:
                 with st.spinner("Generating response..."):
                     try:
@@ -140,7 +140,7 @@ else:
                         output = st.session_state["model"].generate(
                             input_ids=input_ids,
                             attention_mask=attention_mask,
-                            max_length=50,
+                            max_length=250,
                             pad_token_id=st.session_state["tokenizer"].pad_token_id,
                             eos_token_id=st.session_state["tokenizer"].eos_token_id,
                             do_sample=False