Update app.py

app.py

@@ -489,7 +489,7 @@ plt.show()
 #  psnr = tf.image.psnr(Input, Output, max_val=255)
 #  return psnr
   
-def result(Input,Choice):
+def result(Input,Choice,Step):
     if Choice=="Coloring":
       pre_trained = tf.keras.models.load_model("gradio_pix2pix.h5")
       size0 = Input.shape[0]
@@ -513,7 +513,13 @@ def result(Input,Choice):
       return Input,Psnr
       
     if Choice=="Indoor-Coloring":
-      pre_trained = tf.keras.models.load_model("gradio_pix2pix.h5")
+      if Step = 1:
+        pre_trained = tf.keras.models.load_model("indoor_1.h5")
+      if Step = 2:
+        pre_trained = tf.keras.models.load_model("indoor_2.h5")
+      if Step = 3:
+        pre_trained = tf.keras.models.load_model("indoor_1.h5")
+       
       size0 = Input.shape[0]
       size1 = Input.shape[1]
       start = Input
@@ -577,6 +583,6 @@ def result(Input,Choice):
 
 import gradio as gr
 
-iface = gr.Interface(fn=result, inputs=[gr.inputs.Image(type="numpy",image_mode="RGB"),gr.inputs.Radio(["Indoor-Coloring","Outdoor-Coloring","Enhancement","Repair","Repair and Color"])], outputs=[gr.outputs.Image( type="auto", label="Output"),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(["Indoor-Coloring","Outdoor-Coloring","Enhancement","Repair","Repair and Color"]),gradio.inputs.Slider(minimum=1,maximum=3,default=3)], outputs=[gr.outputs.Image( type="auto", label="Output"),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)