Hugging Face's logo

Spaces:
kalinin-a-i
/
ML2_HW4_Kalinin
Running

App Files Community
Alexey Kalinin commited on
Commit
756780e
·
1 Parent(s): 5a69625

fix bug

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ elif input_method == "Manual Input":
41
 
42
  # Classification and output
43
  if title and abstract:
44
- categories = top_95_labels(title, abstract) # Assuming classify returns a list of labels
45
  st.write(f"### Title: {title}")
46
  st.write("**Predicted Categories:**")
47
  for category in categories:
 
41
 
42
  # Classification and output
43
  if title and abstract:
44
+ categories = top_95_labels(pipe, class2name, title, abstract) # Assuming classify returns a list of labels
45
  st.write(f"### Title: {title}")
46
  st.write("**Predicted Categories:**")
47
  for category in categories: