Update app.py

app.py

@@ -70,6 +70,19 @@ HF_TOKEN_SD = os.environ.get('HF_TOKEN_SD')
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print(f'device = {device}')
 
+def get_image_ext(img_bytes):
+    w = imghdr.what("", img_bytes)
+    if w is None:
+        w = "jpeg"
+    return w
+    
+def read_content(file_path):
+    """read the content of target file
+    """
+    with open(file_path, 'rb') as f:
+        content = f.read()
+    return content
+    
 model = None
 
 def model_process(image, mask, alpha_channel, ext):
@@ -171,13 +184,6 @@ transform = transforms.Compose([
 ])
 '''
 
-def read_content(file_path):
-    """read the content of target file
-    """
-    with open(file_path, 'rb') as f:
-        content = f.read()
-
-    return content
 
 image_type = 'filepath' #'pil'
 def predict(input):
@@ -307,17 +313,17 @@ with image_blocks as demo:
         with gr.Box():
             with gr.Row():
                 with gr.Column():
-                    image = gr.Image(source='upload', tool='sketch',type=f'{image_type}', label="Upload").style(height=512)
+                    image = gr.Image(source='upload', elem_id='input_img', tool='sketch', type=f'{image_type}', label="Upload") #.style(height=512)
                     with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
                         # prompt = gr.Textbox(placeholder = 'Your prompt (what you want in place of what is erased)', show_label=False, elem_id="input-text")
-                        btn = gr.Button("Done!").style(
+                        btn_in = gr.Button("Done!").style(
                             margin=True,
                             rounded=(True, True, True, True),
                             full_width=True,
                         )                
                 
                 # with gr.Column():
-                    image_out = gr.Image(label="Output").style(height=512)
+                    image_out = gr.Image(label="Output", elem_id='output_img', visible=False) #.style(height=512)
                     '''
                     with gr.Group(elem_id="share-btn-container"):
                         community_icon = gr.HTML(community_icon_html, visible=False)
@@ -328,7 +334,7 @@ with image_blocks as demo:
                 
 
             # btn.click(fn=predict, inputs=[image, prompt], outputs=[image_out, community_icon, loading_icon, share_button])
-            btn.click(fn=predict, inputs=[image], outputs=[image_out]) #, community_icon, loading_icon, share_button])
+            btn_in.click(fn=predict, inputs=[image], outputs=[image]) #, community_icon, loading_icon, share_button])
             #share_button.click(None, [], [], _js=share_js)