Spaces:
Sleeping
Sleeping
remove limits
Browse files
app.py
CHANGED
@@ -40,7 +40,7 @@ if st.button("Get tags"):
|
|
40 |
st.write(article_text[:500] + "...")
|
41 |
|
42 |
with st.spinner("Analyzing... Please wait."):
|
43 |
-
result = model(article_text
|
44 |
|
45 |
st.write("**Predicted Tags:**")
|
46 |
for label, score in zip(result["labels"], result["scores"]):
|
|
|
40 |
st.write(article_text[:500] + "...")
|
41 |
|
42 |
with st.spinner("Analyzing... Please wait."):
|
43 |
+
result = model(article_text, categories, multi_label=True)
|
44 |
|
45 |
st.write("**Predicted Tags:**")
|
46 |
for label, score in zip(result["labels"], result["scores"]):
|