Update app.py

app.py

@@ -1,34 +1,31 @@
 import streamlit as st
 from transformers import pipeline
 
-# Define the path to the saved model
-model_path = './QAModel'  
-
-# Load the pipeline
-qa_pipeline = pipeline("question-answering", model=model_path, tokenizer=model_path)
+# Load the conversational pipeline
+model_name = "QAModel"
+chatbot_pipeline = pipeline("conversational", model=model_name, tokenizer=model_name)
 
 # Set the title for the Streamlit app
-st.title("Movie Trivia Question Answering")
+st.title("Movie Trivia Chatbot")
 
-# Text inputs for the user
-context = st.text_area("Enter the context (movie-related text):")
-question = st.text_area("Enter your question:")
+# Text input for the user
+user_input = st.text_area("Ask a movie trivia question:")
 
-def generate_answer(question, context):
-    # Perform question answering
-    result = qa_pipeline(question=question, context=context)
-    return result['answer']
+def get_response(user_input):
+    # Generate response
+    conversation = chatbot_pipeline(user_input)
+    return conversation[0]['generated_text']
 
 if st.button("Get Answer"):
-    if context and question:
-        generated_answer = generate_answer(question, context)
-        # Display the generated answer
-        st.subheader("Generated Answer")
-        st.write(generated_answer)
+    if user_input:
+        response = get_response(user_input)
+        # Display the response
+        st.subheader("Answer")
+        st.write(response)
     else:
-        st.warning("Please enter both context and question.")
+        st.warning("Please enter a question.")
 
 # Optionally, add instructions or information about the app
 st.write("""
-    Enter a movie-related context and a question related to the context above. The model will provide the answer based on the context provided.
+    Enter a movie-related question above. The chatbot will provide an answer based on its training.
     """)