alirani commited on
Commit
5e605db
1 Parent(s): 3894141

fix generate classification

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ if functionality == "Classification" :
50
 
51
  if button_classify:
52
  if (len(prod_title.split(' ')) > 0) & len(prod_synopsis.split(' ')) > 0:
53
- classified_genre = generate_classification(model_gen, tokenizer_gen, prod_title, prod_synopsis)
54
  st.write('The genre of the title & synopsis is : ', classified_genre)
55
  else:
56
  st.write('Write a title & synopsis for the classifier to work !')
 
50
 
51
  if button_classify:
52
  if (len(prod_title.split(' ')) > 0) & len(prod_synopsis.split(' ')) > 0:
53
+ classified_genre = generate_classification(model_clf, tokenizer_clf, prod_title, prod_synopsis)
54
  st.write('The genre of the title & synopsis is : ', classified_genre)
55
  else:
56
  st.write('Write a title & synopsis for the classifier to work !')