TAgroup5 commited on
Commit
c34ca4d
·
verified ·
1 Parent(s): 3ee91ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- qa_pipeline = pipeline("question-answering", model=model, tokenizer=tokenizer)
 
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")