code

app.py

@@ -49,7 +49,7 @@ def main():
 
     with gr.Blocks(theme=args.theme, css='style.css') as demo:
         gr.Markdown('''# World Puzzle Solver 🧩''')
-        gr.Markdown('''## (Works in Spanish too!) 🇪🇸''')
+        gr.Markdown('''## (Works in Spanish too!)  🇪🇸''')
 
         with gr.Box():
             gr.Markdown(
@@ -68,7 +68,7 @@ def main():
                         with gr.Row():
                             input_words = gr.Image(label='Words',
                                                    type='filepath',
-                                                   interactive=True, height="300px", width="300px",
+                                                   interactive=True,
                                                    )
                         with gr.Row():
                             crop_words_button = gr.Button('Crop Words ✂️')
@@ -82,18 +82,18 @@ def main():
                                                         label='Image Examples (Drag and drop into both boxes) then crop using the tool button')
 
                 with gr.Box():
+                    # Change column height
+                    
                     with gr.Column():
                         gr.Markdown('''Cropped Images ✂️''')
                         with gr.Row():
                             cropped_board = gr.Image(label='Board Cropped',
                                                      type='filepath',
                                                      interactive=False, height="auto")
-                            instyle = gr.Variable()
                         with gr.Row():
                             cropped_words = gr.Image(label='Words Cropped',
                                                      type='filepath',
                                                      interactive=False)
-                            instyle = gr.Variable()
                         with gr.Row():
                             find_words_button = gr.Button('Find Words 🔍')
                         with gr.Row():

solver.py

@@ -19,7 +19,7 @@ with open("class_names.txt", "r") as f:  # reading them in from class_names.txt
 
 model1 = tf.keras.models.load_model('model/model30.h5')
 model2 = tf.keras.models.load_model('model/model15.h5')
-#model3 = tf.keras.models.load_model('model/model2.h5')
+model3 = tf.keras.models.load_model('model/model2.h5')
 
 palabras_1 = []
 # Borrar el directorio de imagenes
@@ -177,12 +177,18 @@ def solve_puzzle(img, words):
                 img_array = img2.reshape(1, 28, 28, 1)
                 prediction1 = np.argmax(model1.predict(img_array))
                 prediction2 = np.argmax(model2.predict(img_array))
-                #prediction3 = np.argmax(model3.predict(img_array))
+                prediction3 = np.argmax(model3.predict(img_array))
                 pred = 0
-                if prediction1 == prediction2:
+                if prediction1 == prediction2 and prediction2 == prediction3:
                    pred = prediction1
+                elif  prediction1 == prediction2:
+                   pred = prediction1
+                elif  prediction2 == prediction3:
+                   pred = prediction2
+                elif  prediction1 == prediction3:
+                   pred = prediction3
                 else:
-                    pred = 32
+                   pred = 32
                 #print(characters[pred])
                 contCuadrados["anchura"] = x
                 contCuadrados["altura"] = y