Update app.py

app.py

@@ -492,11 +492,11 @@ plt.show()
 def result(Input,Choice,Step):
       
     if Choice=="Indoor-Coloring":
-      if Step == 1:
+      if Step == 1.0:
         pre_trained = tf.keras.models.load_model("indoor_1.h5")
-      if Step == 2:
+      if Step == 2.0:
         pre_trained = tf.keras.models.load_model("indoor_2.h5")
-      if Step == 3:
+      if Step == 3.0:
         pre_trained = tf.keras.models.load_model("indoor_3.h5")
        
       size0 = Input.shape[0]
@@ -521,11 +521,11 @@ def result(Input,Choice,Step):
       
     if Choice=="Outdoor-Coloring":
     
-      if Step == 1:
+      if Step == 1.0:
         pre_trained = tf.keras.models.load_model("outdoor_1.h5")
-      if Step == 2:
+      if Step == 2.0:
         pre_trained = tf.keras.models.load_model("outdoor_2.h5")
-      if Step == 3:
+      if Step == 3.0:
         pre_trained = tf.keras.models.load_model("outdoor_3.h5")
         
       size0 = Input.shape[0]