Update app.py

app.py

@@ -1,11 +1,16 @@
 import streamlit as st
 from transformers import pipeline
 
-# Load the model and tokenizer from the models/ directory
+from transformers import AutoTokenizer, AutoModelForQuestionAnswering
+
+model_name = "wedo2910/qa_arabic_model"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForQuestionAnswering.from_pretrained(model_name)
+
 qa_pipeline = pipeline(
     "question-answering",
-    model="wedo2910/qa_arabic_model",
-    tokenizer="wedo2910/qa_arabic_model"
+    model=model,
+    tokenizer=tokenizer
 )
 
 # Default settings