Update app.py

app.py

@@ -10,6 +10,15 @@ import gradio as gr
 # Declaring Constants
 SAVED_MODEL_PATH = "https://tfhub.dev/captain-pool/esrgan-tf2/1"
 
+def resize(width,img):
+  basewidth = width
+  img = Image.open(img)
+  wpercent = (basewidth/float(img.size[0]))
+  hsize = int((float(img.size[1])*float(wpercent)))
+  img = img.resize((basewidth,hsize), Image.ANTIALIAS)
+  img.save('somepic.jpg')
+  return 'somepic.jpg'
+  
 def preprocess_image(image_path):
   """ Loads image from path and preprocesses to make it model ready
       Args:
@@ -40,11 +49,12 @@ def plot_image(image):
   
 model = hub.load(SAVED_MODEL_PATH)
 def inference(img):
-  hr_image = preprocess_image(img)
+  resize_image = resize(256,img)
+  hr_image = preprocess_image(resize_image)
   fake_image = model(hr_image)
   fake_image = tf.squeeze(fake_image)
   pil_image = plot_image(tf.squeeze(fake_image))
   return pil_image
   
-gr.Interface(inference,gr.inputs.Image(type="filepath",shape=(256,256)),"image").launch(enable_queue=True)
+gr.Interface(inference,gr.inputs.Image(type="filepath"),"image").launch(enable_queue=True)