Update app.py

app.py

@@ -1,14 +1,9 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the pre-trained BART model and tokenizer
-bart_model_path = "bart_QA"
-if os.path.exists(bart_model_path):
-    bart_model = AutoModelForSeq2SeqLM.from_pretrained(bart_model_path)
-    print(f"Loaded BART model from {bart_model_path}")
-else:
-    print(f"BART model not found in {bart_model_path}. Please ensure the model is uploaded.")
-    exit(1)
+# Load the trained model
+model_path = "bart_QA"
+qa_pipeline = pipeline("question-answering", model=model_path, tokenizer=model_path)
 
 def answer_question(question, context):
     if question.strip() == "" or context.strip() == "":