fix

app.py

@@ -32,7 +32,6 @@ st.title('Demo LoreFinder')
 st.header('Generate a story')
 
 prod_title = st.text_input('Type a title to generate a synopsis')
-prod_synopsis = st.text_input('Type a synopsis to guess the genre')
 
 option_genres = st.selectbox(
     'Select a genre to tailor your synopsis',
@@ -43,19 +42,9 @@ option_genres = st.selectbox(
 
 button_synopsis = st.button('Get synopsis')
 
-button_genres = st.button('Get genres')
-
 if button_synopsis:
     if len(prod_title.split(' ')) > 0:
         gen_synopsis = generate_synopsis(model_gen, tokenizer_gen, f"{prod_title} | {option_genres} | ")
         st.text_area('Generated synopsis', value=gen_synopsis, disabled=True)
     else:
-        st.write('Write a title for the generator to work !')
-
-if button_genres:
-    if len(prod_title.split(' ')) > 0:
-        classifier_output = generate_classification(model_clf, tokenizer_clf, prod_title, prod_synopsis)
-        print(classifier_output)
-        st.write(f"Guessed genre : {classifier_output}")
-    else:
-        st.write('Write a title and a synopsis for the classifier to work !')
+        st.write('Write a title for the generator to work !')