diff --git a/webui.py b/webui.py
index 00c0725..81b1bed 100644
--- a/webui.py
+++ b/webui.py
@@ -1,7 +1,7 @@
 import argparse, os, sys, glob, re
 
 from frontend.frontend import draw_gradio_ui
-
+from frontend.ui_functions import resize_image
 parser = argparse.ArgumentParser()
 parser.add_argument("--outdir", type=str, nargs="?", help="dir to write results to", default=None)
 parser.add_argument("--outdir_txt2img", type=str, nargs="?", help="dir to write txt2img results to (overrides --outdir)", default=None)
diff --git a/webui_playground.py b/webui_playground.py
index 2012d88..35b6b58 100644
--- a/webui_playground.py
+++ b/webui_playground.py
@@ -7,14 +7,14 @@ import base64
 import re
 
 from frontend.frontend import draw_gradio_ui
+from frontend.ui_functions import resize_image
 
 """
 This file is here to play around with the interface without loading the whole model 
 
 TBD - extract all the UI into this file and import from the main webui. 
 """
-def resize_image(resize_mode, im, width, height):
-    return im 
+
 GFPGAN = True
 RealESRGAN = True 
 def run_goBIG():