Update app_gradio/app.py

app_gradio/app.py

@@ -11,7 +11,7 @@ from PIL import ImageStat
 from PIL.Image import Image
 import requests
 
-from image_to_fen.fen import ImageToFen
+from image_to_fen.fen import ImageToFen, fen_and_image
 import image_to_fen.util as util
 
 os.environ["CUDA_VISIBLE_DEVICES"] = ""  # do not use GPU
@@ -78,7 +78,8 @@ class PredictorBackend:
             self._predict = self._predict_from_endpoint
         else:
             model = ImageToFen()
-            self._predict = model.predict
+            # self._predict = model.predict
+            self._predict = fen_and_image
 
     def run(self, image):
         pred, metrics = self._predict_with_metrics(image)