app.py

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

#qa_pipeline = pipeline("question-answering")
qa_pipeline = pipeline("question-answering", model="microsoft/Phi-3-mini-4k-instruct")
st.title("Adrega AI Help")
dataset = load_dataset("andreska/adregadocs", split="test")
context = dataset[1]["text"]



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