ksvmuralidhar
commited on
Commit
•
2b4981e
1
Parent(s):
3f9aa0e
Update app.py
Browse files
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
|
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)
|