Update app.py

app.py

@@ -1,8 +1,11 @@
-import streamlit as st
-from transformers import pipeline
+from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
 
-# Load the question-answering pipeline
-qa_pipeline = pipeline("question-answering", model="nlpaueb/bert-base-greek-uncased-v1")
+# Load model and tokenizer
+model = AutoModelForQuestionAnswering.from_pretrained("spyrosbriakos/greek_legal_bert_v2")
+tokenizer = AutoTokenizer.from_pretrained("spyrosbriakos/greek_legal_bert_v2")
+
+# Create QA pipeline
+qa_pipeline = pipeline("question-answering", model=model, tokenizer=tokenizer)
 
 # Title of the app
 st.title("Question Answering in Greek")