chap0lin commited on
Commit
646a2f5
1 Parent(s): cb6b313

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -52,7 +52,7 @@ reconstructed_model_CNN = keras.models.load_model("best weights CNN.h5",
52
  "recall_m":recall_m})
53
 
54
 
55
- def greet(sentence):
56
  sentenceWords = json.loads(sentence.replace("'",'"'))
57
 
58
  aux_vector = []
@@ -67,9 +67,9 @@ def greet(sentence):
67
  value = reconstructed_model_CNN.predict(MCTIinput_vector)[0]
68
 
69
  if value >= 0.5:
70
- return gr.Image(source = "elegivel.png")
71
  else:
72
- return gr.Image(source = "inelegivel.png")
73
 
74
- iface = gr.Interface(fn=greet, inputs="text", outputs="image")
75
  iface.launch()
 
52
  "recall_m":recall_m})
53
 
54
 
55
+ def classify(sentence):
56
  sentenceWords = json.loads(sentence.replace("'",'"'))
57
 
58
  aux_vector = []
 
67
  value = reconstructed_model_CNN.predict(MCTIinput_vector)[0]
68
 
69
  if value >= 0.5:
70
+ return gr.Image("elegivel.png")
71
  else:
72
+ return gr.Image("inelegivel.png")
73
 
74
+ iface = gr.Interface(fn=classify, inputs="text", outputs="image")
75
  iface.launch()