Update app.py

app.py

@@ -117,26 +117,14 @@ def sepia(input_img):
 
     fig = draw_plot(pred_img, seg)
 
-    tf.keras.backend.clear_session()
     
     return fig
 
-    
-class TensorFlowImageInput(gr.Component):
-    def __init__(self, shape=(400, 600)):
-        super().__init__()
-        self.image = None
-        self.shape = shape
-
-    def render(self):
-        return gr.File(label="이미지", accept=".jpg,.png", capture_file=True)
-
-    def update(self, value):
-        self.image = tf.image.decode_jpeg(value)
-
-    def get_value(self):
-        return self.image
+demo = gr.Interface(fn=sepia,
+                    inputs=gr.Image(shape=(400, 600)),
+                    outputs=['plot'],
+                    examples=["side-1.jpg", "side-2.jpg", "side-3.jpg", "side-4.jpg"],
+                    allow_flagging='never')
 
-demo = gr.Interface(fn=sepia, inputs=[TensorFlowImageInput(shape=(400, 600))], outputs=["plot"])
-demo.launch()
 
+demo.launch()