Update app.py

app.py

@@ -1,15 +1,29 @@
 import gradio as gr
 from gensim.models import Word2Vec
+from gensim.models.callbacks import CallbackAny2Vec
 import keras
 from keras.utils.data_utils import pad_sequences
 import requests
 import shutil
 import requests
 
-def get_model(filename, url):
-    response = requests.get(url, stream=True)
-    with open(filename, 'wb') as fin:
-        shutil.copyfileobj(response.raw, fin)
+#initialise callback class
+class callback(CallbackAny2Vec):
+  """
+  Print the loss value after each epoch
+  """
+  def __init__(self):
+    self.epoch = 0
+    #gensim loss is cumulative, so we record previous values to print
+    self.loss_previous_step = 0 
+
+  def on_epoch_end(self, model):
+    loss = model.get_latest_training_loss()
+    if self.epoch % 100 == 0:
+      print('Loss after epoch {}: {}'.format(self.epoch, loss-self.loss_previous_step))
+
+    self.epoch+= 1
+    self.loss_previous_step = loss
 
 # get_model("word2vec_xp8.model", "https://github.com/chap0lin/PPF-MCTI/blob/master/Meta10/pesos/word2vec/word2vec_xp8.model?raw=true")
 reloaded_w2v_model = Word2Vec.load('word2vec_xp8.model')