Update app.py

app.py

@@ -44,11 +44,8 @@ class callback(CallbackAny2Vec):
     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')
 
-# get_model("best weights CNN.h5", "https://github.com/chap0lin/PPF-MCTI/blob/master/Meta10/pesos/word2vec/best%20weights%20CNN.h5?raw=true")
-
 reconstructed_model_CNN = keras.models.load_model("best weights CNN.h5", 
                                                    custom_objects={'f1_m':f1_m, 
                                                                    "precision_m":precision_m, 
@@ -67,8 +64,12 @@ def greet(sentence):
   
 
   # pad 2726
+  value = reconstructed_model_CNN.predict(MCTIinput_vector)[0]
   
-  return reconstructed_model_CNN.predict(MCTIinput_vector)
+  if value >= 0.5:
+    return gr.Image(source = "elegivel.png")
+  else:
+    return gr.Image(source = "inelegivel.png")
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+iface = gr.Interface(fn=greet, inputs="text", outputs="image")
 iface.launch()