Update app.py

app.py

@@ -481,13 +481,13 @@ plt.show()
 #plt.imsave(name,pre)
 #cv2.imshow(pre)
 
-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 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,Output,Choice):
-  psnr = tf.image.psnr(Input, Output, max_val=255)
-  return psnr
+#def psnr(Input,Output,Choice):
+#  psnr = tf.image.psnr(Input, Output, max_val=255)
+#  return psnr
   
 def result(Input,Choice):
     if Choice=="Coloring":
@@ -502,7 +502,7 @@ def result(Input,Choice):
       Output = (Output+1)*127.5
       Output = np.uint8(Output)
       Output = cv2.resize(Output, (size1,size0), interpolation = cv2.INTER_AREA)
-      psnr = psnr(Input,Output,"Coloring")
+      psnr = psnr = tf.image.psnr(Input, Output, max_val=255)
       return Output,psnr
       
     if Choice=="Enhancement":
@@ -526,6 +526,6 @@ def result(Input,Choice):
 
 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"])], outputs=[gr.outputs.Image( type="auto", label=None),gr.outputs.Textbox(type="number",label="psnr")],theme="grass",live=True
+iface = gr.Interface(fn=result, 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)