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()