test3

app.py

@@ -62,7 +62,7 @@ def main():
         user_question = st.text_input("Ask a Question from the document")
         if user_question:
             if not qa_pipeline and "raw_text" in st.session_state:
-                model_name = "notabaka/DocQA"
+                model_name = "HanNayeoniee/LHK_DPO_v1"
                 tokenizer = AutoTokenizer.from_pretrained(model_name)
                 model = AutoModelForQuestionAnswering.from_pretrained(model_name)
                 qa_pipeline = pipeline("question-answering", model=model, tokenizer=tokenizer)