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)
|