import streamlit as st
from transformers import pipeline
#from datasets import load_dataset

#qa_pipeline = pipeline("question-answering")
qa_pipeline = pipeline("question-answering")
st.title("Adrega AI Help")
#dataset = load_dataset("andreska/adregadocs", split="test")
context = "Adrega is a very cool company, that implements AI" #dataset[0]["text"]



user_input = st.text_input('Ask me a question')
if st.button("Submit"): 
    if user_input: 
        result = qa_pipeline(question=user_input, context=context) 
        st.write(f"Adrega AI: {result['answer']}") 
    else: 
        st.write("Please enter a question.")