Update app.py

app.py

@@ -6,7 +6,7 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipe
 from transformers import AutoModelForQuestionAnswering
 
 # Load fine-tuned models and tokenizers for both functions
-model_name_classification = "TAgroup5/news-classification-model"  # Replace with the correct model name
+model_name_classification = "TAgroup5/news-classification-model"
 model = AutoModelForSequenceClassification.from_pretrained(model_name_classification)
 tokenizer = AutoTokenizer.from_pretrained(model_name_classification)
 
@@ -15,6 +15,14 @@ model_qa = AutoModelForQuestionAnswering.from_pretrained(model_name_qa)
 tokenizer_qa = AutoTokenizer.from_pretrained(model_name_qa)
 
 # Initialize pipelines
+from transformers import pipeline, AutoModelForQuestionAnswering, AutoTokenizer
+
+model_name = "distilbert-base-cased-distilled-squad"  # Example model
+model = AutoModelForQuestionAnswering.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+qa_pipeline = pipeline("question-answering", model=model, tokenizer=tokenizer)
+
 text_classification_pipeline = pipeline("text-classification", model=model, tokenizer=tokenizer)
 qa_pipeline = pipeline("question-answering", model=model)