Spaces:
Runtime error
Runtime error
alirani
commited on
Commit
•
5e605db
1
Parent(s):
3894141
fix generate classification
Browse files
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(
|
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 !')
|