Update app.py

app.py

@@ -11,14 +11,14 @@ os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
 
 
 # Create an instance of the model
-CALTEXT = CALTextModel.CALTEXT_Model(training=False)
-CALTEXT.load_weights('final_caltextModel/cp-0037.ckpt')
+CALText = CALTextModel.CALText_Model(training=False)
+CALText.load_weights('final_caltextModel/cp-0037.ckpt')
 test_loss = tf.keras.metrics.Mean(name='test_loss')
 
 
 def recognize_text(input_image):
     x, x_mask=data.preprocess_img(input_image)
-    output_str, gifImage=CALTextModel.predict(x, x_mask)
+    output_str, gifImage=CALTextModel.predict(CALText, x, x_mask)
     return output_str,gifImage
 
 examples = [['sample_test_images/59-11.png'],