Update app.py

app.py

@@ -7,12 +7,7 @@ import tensorflow_hub as hub
 import matplotlib.pyplot as plt
 import gradio as gr
 
-os.environ["TFHUB_DOWNLOAD_PROGRESS"] = "True"
-
-os.system("wget https://user-images.githubusercontent.com/12981474/40157448-eff91f06-5953-11e8-9a37-f6b5693fa03f.png -O original.png")
-
 # Declaring Constants
-IMAGE_PATH = "original.png"
 SAVED_MODEL_PATH = "https://tfhub.dev/captain-pool/esrgan-tf2/1"
 
 def preprocess_image(image_path):
@@ -46,12 +41,10 @@ def plot_image(image):
 model = hub.load(SAVED_MODEL_PATH)
 def inference(img):
   hr_image = preprocess_image(img)
-  start = time.time()
   fake_image = model(hr_image)
   fake_image = tf.squeeze(fake_image)
-  print("Time Taken: %f" % (time.time() - start))
   pil_image = plot_image(tf.squeeze(fake_image))
   return pil_image
   
-gr.Interface(inference,gr.inputs.Image(type="filepath"),"image").launch(enable_queue=True)
+gr.Interface(inference,gr.inputs.Image(type="filepath",shape=(256,256)),"image").launch(enable_queue=True)