100 epochs

app.py

@@ -147,7 +147,7 @@ cp_callback = tf.keras.callbacks.ModelCheckpoint(
     save_weights_only=True,
     save_freq=2)
 
-model.fit(cached_train, callbacks=[cp_callback],epochs=2)
+model.fit(cached_train, callbacks=[cp_callback],epochs=100)
 
 index=df["code"].map(lambda x: [model.movie_model(tf.constant(x))])
 
@@ -177,6 +177,9 @@ def predict(text):
       plt.text(x, y, y, ha='center', va='bottom', fontsize=12, color='black')
     return xx, fig
 
+predict("")
+
+
 demo = gr.Interface(fn=predict, inputs=gr.inputs.Textbox(label='CANDIDATE COMPETENCES - Click *Clear* before adding new input'), \
     outputs=[gr.outputs.Textbox(label='SUGGESTED VACANCIES'),\
       gr.Plot()],\