Update app.py

app.py

@@ -9,32 +9,33 @@ st.write("Ask any question, and WhiteRabbitNEO will provide an answer.")
 @st.cache_resource
 def load_model():
     try:
-        # Replace with the correct model if needed (e.g., ensure WhiteRabbitNEO exists)
+        # Simple model pipeline initialization
         model = pipeline("question-answering", model="WhiteRabbitNEO")
         return model
     except Exception as e:
         st.error(f"Failed to load model: {e}")
         return None
 
+# Load the model
 qa_pipeline = load_model()
 
-# Input Section
+# Simple input section: Ask question and provide context
 if qa_pipeline:
-    question = st.text_input("Enter your question:")
-    context = st.text_area("Provide some context for the question:")
+    question = st.text_input("Your question:")
+    context = st.text_area("Context (provide background info for the question):")
 
-    # Output Section
+    # Button to trigger the model for prediction
     if st.button("Get Answer"):
         if question and context:
-            with st.spinner("Thinking..."):
-                try:
-                    result = qa_pipeline(question=question, context=context)
-                    st.success("Answer:")
-                    st.write(result["answer"])
-                    st.write(f"Confidence: {result['score']:.2f}")
-                except Exception as e:
-                    st.error(f"Error generating answer: {e}")
+            try:
+                # Get prediction from the model
+                result = qa_pipeline(question=question, context=context)
+                # Display the answer
+                st.write(f"**Answer:** {result['answer']}")
+                st.write(f"**Confidence Score:** {result['score']:.2f}")
+            except Exception as e:
+                st.error(f"Error generating answer: {e}")
         else:
-            st.warning("Please provide both a question and context.")
+            st.warning("Please fill both the question and context fields.")
 else:
-    st.error("Failed to load the question-answering model. Please try again later.")
+    st.error("Model could not be loaded. Please check your configuration.")