Update app.py

app.py

@@ -122,14 +122,21 @@ def sepia(input_img):
     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)
 
-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')
-
+    def get_value(self):
+        return self.image
 
+demo = gr.Interface(fn=sepia, inputs=[TensorFlowImageInput(shape=(400, 600))], outputs=["plot"])
 demo.launch()
+