Reinserted question-answering instead of text-generation

app.py

@@ -2,8 +2,8 @@ import streamlit as st
 from transformers import pipeline
 #from datasets import load_dataset
 
-#pipe = pipeline("question-answering")
-pipe = pipeline("text-generation")
+pipe = pipeline("question-answering")
+#pipe = pipeline("text-generation")
 
 st.title("Adrega AI Help")
 #dataset = load_dataset("andreska/adregadocs", split="test")
@@ -12,9 +12,11 @@ context = "Adrega is a very cool company, that implements AI. Rett fra Rio is a
 user_input = st.text_input('Ask me a question')
 if st.button("Submit"): 
     if user_input: 
-        text_inputs = f"Context: {context}\nQuestion: {user_input}\nAnswer:"
-        result = pipe(text_inputs, max_length=200, num_return_sequences=1)[0]['generated_text']
-        answer = result.split("Answer:")[1].strip()
+        #text_inputs = f"Context: {context}\nQuestion: {user_input}\nAnswer:"
+        #result = pipe(text_inputs, max_length=200, num_return_sequences=1)[0]['generated_text']
+        #answer = result.split("Answer:")[1].strip()
+
+        answer = pipe(question=user_input, context=context)
         st.write(f"Adrega AI: {answer}")
     else: 
         st.write("Please enter a question.")