ksvmuralidhar commited on
Commit
2b4981e
1 Parent(s): 3f9aa0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def inference(text):
39
  def main():
40
  st.title("News Headline Classification")
41
  lang_trained = 'TECHNOLOGY, HEALTH, WORLD, ENTERTAINMENT, SPORTS, BUSINESS, NATION, SCIENCE'
42
- st.write(f'Model is trained on the following categories \n{lang_trained}')
43
  review = st.text_area("Enter a news headline:", "", height=100)
44
  if st.button("Submit"):
45
  result = inference(review)
 
39
  def main():
40
  st.title("News Headline Classification")
41
  lang_trained = 'TECHNOLOGY, HEALTH, WORLD, ENTERTAINMENT, SPORTS, BUSINESS, NATION, SCIENCE'
42
+ st.write(f'Model is fine-tuned on the following categories \n{lang_trained}')
43
  review = st.text_area("Enter a news headline:", "", height=100)
44
  if st.button("Submit"):
45
  result = inference(review)