TAgroup5 commited on
Commit
6fcbed4
Β·
verified Β·
1 Parent(s): 4a1b338

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -52,7 +52,7 @@ model_name_qa = "distilbert-base-cased-distilled-squad"
52
  model_qa = AutoModelForQuestionAnswering.from_pretrained(model_name_qa)
53
  tokenizer_qa = AutoTokenizer.from_pretrained(model_name_qa)
54
 
55
- # Initialize pipelines
56
  text_classification_pipeline = pipeline("text-classification", model=model, tokenizer=tokenizer)
57
  qa_pipeline = pipeline("question-answering", model=model_qa, tokenizer=tokenizer_qa)
58
 
@@ -89,6 +89,7 @@ if uploaded_file is not None:
89
  df.to_csv(output, index=False, encoding="utf-8-sig")
90
  st.download_button("πŸ“₯ Download Classified News", data=output.getvalue(), file_name="output.csv", mime="text/csv")
91
 
 
92
  st.write("πŸ” **Filter by Category**")
93
  categories = ['All', 'Business', 'Opinion', 'Political_gossip', 'Sports', 'World_news']
94
 
 
52
  model_qa = AutoModelForQuestionAnswering.from_pretrained(model_name_qa)
53
  tokenizer_qa = AutoTokenizer.from_pretrained(model_name_qa)
54
 
55
+ # Initialize pipelines for both models
56
  text_classification_pipeline = pipeline("text-classification", model=model, tokenizer=tokenizer)
57
  qa_pipeline = pipeline("question-answering", model=model_qa, tokenizer=tokenizer_qa)
58
 
 
89
  df.to_csv(output, index=False, encoding="utf-8-sig")
90
  st.download_button("πŸ“₯ Download Classified News", data=output.getvalue(), file_name="output.csv", mime="text/csv")
91
 
92
+ #App Component 3: Think!Think!Think! - Introducing a News Filtering Option
93
  st.write("πŸ” **Filter by Category**")
94
  categories = ['All', 'Business', 'Opinion', 'Political_gossip', 'Sports', 'World_news']
95