herztard commited on
Commit
b7edde4
·
1 Parent(s): cc238ea
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -39,6 +39,9 @@ if st.button("Get tags"):
39
  st.write("**Extracted Article Text:**")
40
  st.write(article_text[:500] + "...")
41
 
42
-
 
 
 
43
  else:
44
  st.write("Please enter a valid news URL.")
 
39
  st.write("**Extracted Article Text:**")
40
  st.write(article_text[:500] + "...")
41
 
42
+ result = model(article_text, categories, multi_label=True)
43
+ st.write("**Predicted Tags:**")
44
+ for label, score in zip(result["labels"], result["scores"]):
45
+ st.write(f"- {label}: {score:.2f}")
46
  else:
47
  st.write("Please enter a valid news URL.")