alirani commited on
Commit
c374e34
·
1 Parent(s): 9f4fb0e
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -55,6 +55,7 @@ if button_synopsis:
55
  if button_genres:
56
  if len(prod_title.split(' ')) > 0:
57
  classifier_output = generate_classification(model_clf, tokenizer_clf, prod_title, prod_synopsis)
 
58
  st.write(f"Guessed genre : {classifier_output}")
59
  else:
60
  st.write('Write a title and a synopsis for the classifier to work !')
 
55
  if button_genres:
56
  if len(prod_title.split(' ')) > 0:
57
  classifier_output = generate_classification(model_clf, tokenizer_clf, prod_title, prod_synopsis)
58
+ print(classifier_output)
59
  st.write(f"Guessed genre : {classifier_output}")
60
  else:
61
  st.write('Write a title and a synopsis for the classifier to work !')