Update app.py

app.py

@@ -514,26 +514,18 @@ 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):
-#  Output = result(Input,Choice)
-#  psnr = tf.image.psnr(Input, Output, max_val=255)
-#  return psnr
-
-def psnr(Input, Choice, MAX=None):
-    Output = result(Input,Choice)
-    if MAX is None:
-		    MAX = np.iinfo(Input.dtype).max
-    mse = np.mean((Input - Output) ** 2)
-    if mse == 0:
-        return 100
-    return 20 * math.log10(MAX / math.sqrt(mse))
-    
+def psnr(Input,Choice):
+  Output = result(Input,Choice)
+  psnr = tf.image.psnr(Input, Output, max_val=255)
+  print(psnr)
+  return psnr
+
 #lst = cv2.imread('/content/drive/MyDrive/ColabNotebooks/enhance/low-sat.jpg')
 #r = result(lst)
 #cv2.imshow(r)
 
 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"]),gradio.inputs.Slider(self, 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"]),gr.inputs.Slider(self, 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
 ,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="False",show_tips="True",inbrowser=True)
+iface.launch(debug="True",show_tips="True",inbrowser=True)