Update app.py

app.py

@@ -27,11 +27,14 @@ if user_input:
     
     # Generate output with attention mask and pad token ID
     try:
-        outputs = model.generate(input_ids, attention_mask=attention_mask, max_length=50)
-        st.write("Model Output (Raw):", outputs)  # Debugging model raw output
+        # Increased max_length to 100 for more space for generation
+        outputs = model.generate(input_ids, attention_mask=attention_mask, max_length=100, num_return_sequences=1)
+        
+        # Debugging model raw output (just the token ids)
+        st.write("Model Output (Raw Token IDs):", outputs)
         
         # Decode the output and display
-        answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        st.write("Answer:", answer)
+        decoded_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        st.write("Decoded Answer:", decoded_output)
     except Exception as e:
-        st.write("Error generating output:", str(e))
+        st.write("Error generating output:", str(e))