Update app.py

app.py

@@ -14,7 +14,6 @@ import numpy as np
 examples = [['sample1.png'],
             ['sample2.png']]
 
-inference_on = ['Full Resolution Image', 'Downsampled Image']
 
 title = "Restormer"
 description = """
@@ -27,13 +26,11 @@ Gradio demo for <b>Restormer: Efficient Transformer for High-Resolution Image Re
 article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2111.09881'>Restormer: Efficient Transformer for High-Resolution Image Restoration </a> | <a href='https://github.com/swz30/Restormer'>Github Repo</a></p>"
 
 
-def inference(img, task, run_on):
+def inference(img, task):
     if not os.path.exists('temp'):
       os.system('mkdir temp')
       
-    if run_on == 'Full Resolution Image':
-      img = img
-    else: # 'Downsampled Image'
+    # 'Downsampled Image'
     ####  Resize the longer edge of the input image
       max_res = 512
       width, height = img.size
@@ -86,7 +83,6 @@ gr.Interface(
     [
         gr.Image(type="pil", label="Input"),
         gr.Radio(["Deraining"], default="Denoising", label='task'),
-        gr.Dropdown(choices=inference_on, type="value", default='Downsampled Image', label='Inference on')
 
     ],
     gr.Image(type="pil", label="cleaned and restored"),