Spaces:
Sleeping
Sleeping
show tags
Browse files
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.")
|