Update app.py

app.py

@@ -514,7 +514,7 @@ def ssim(original,predict):
    ssim = tf.image.ssim(original, predict, max_val=1.0, filter_size=11, filter_sigma=1.5, k1=0.01, k2=0.03)
    return ssim
 
-def psnr(Input,Choice,Step):
+def psnr(Input,Choice):
   Output = result(Input,Choice)
   psnr = tf.image.psnr(Input, Output, max_val=255)
   print(psnr)
@@ -526,6 +526,6 @@ def psnr(Input,Choice,Step):
 
 import gradio as gr
 
-iface = gr.Interface(fn=[result,psnr], inputs=[gr.inputs.Image(type="numpy",image_mode="RGB"),gr.inputs.Radio(["Coloring","Enhancement","Repair","Repair and Color"]),gr.inputs.Slider(minimum=0, maximum=300, step=50, default=300)], outputs=[gr.outputs.Image( type="auto", label=None),gr.outputs.Textbox(type="number",label="psnr")],theme="grass",live=True
+iface = gr.Interface(fn=[result,psnr], inputs=[gr.inputs.Image(type="numpy",image_mode="RGB"),gr.inputs.Radio(["Coloring","Enhancement","Repair","Repair and Color"])], outputs=[gr.outputs.Image( type="auto", label=None),gr.outputs.Textbox(type="number",label="psnr")],theme="grass",live=True
 ,css=""" body {background-color: rgba(127,191,63,0.48)} """,title="Image Enhancement",article=""" <a href="https://docs.google.com/document/d/19k6dyR5x_hd1M0yoU8i49dlDWvFmtnBT/edit?usp=sharing&ouid=115743073712072785012&rtpof=true&sd=true" download="example.docx"><img src="https://img.icons8.com/external-itim2101-lineal-color-itim2101/64/000000/external-article-blogger-and-influencer-itim2101-lineal-color-itim2101-1.png" alt="Article"></a>""",examples=[["dog.jpg","Coloring"],["woman.png","Coloring"]]) 
 iface.launch(debug="True",show_tips="True",inbrowser=True)