Update app.py

app.py

@@ -29,6 +29,16 @@ if st.button("Get Response"):
     if user_input:
         # Query the model
         output = query({"inputs": user_input})
-        st.write(f"**Response:** {output['generated_text']}")
+        
+        # Print the entire output for debugging purposes
+        st.write("**API Response:**", output)
+        
+        # Check the structure of the response and adjust this line accordingly
+        if 'generated_text' in output:
+            st.write(f"**Response:** {output['generated_text']}")
+        elif isinstance(output, list) and len(output) > 0:
+            st.write(f"**Response:** {output[0].get('generated_text', 'No response available.')}")
+        else:
+            st.write("**Response:** Unable to retrieve a valid response.")
     else:
         st.write("Please enter a query to get a response.")