Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -26,7 +26,8 @@ tokenizer = AutoTokenizer.from_pretrained("TAgroup5/news-classification-model")
|
|
26 |
|
27 |
# Initialize pipelines
|
28 |
text_classification_pipeline = pipeline("text-classification", model=model, tokenizer=tokenizer)
|
29 |
-
|
|
|
30 |
|
31 |
# Streamlit App
|
32 |
st.title("News Classification and Q&A")
|
|
|
26 |
|
27 |
# Initialize pipelines
|
28 |
text_classification_pipeline = pipeline("text-classification", model=model, tokenizer=tokenizer)
|
29 |
+
qa_model_name = "distilbert-base-uncased-distilled-squad" # Example of a common Q&A model
|
30 |
+
qa_pipeline = pipeline("question-answering", model=qa_model_name, tokenizer=qa_model_name)
|
31 |
|
32 |
# Streamlit App
|
33 |
st.title("News Classification and Q&A")
|