Update app.py

app.py

@@ -4,6 +4,7 @@ import os
 
 def generate(image, prompt, seed):
     print(image, prompt, seed)
+    image_name = os.path.splitext(image)[0]
     command = f"python handrefiner.py --input_img {image} --out_dir output --strength 0.55 --weights models/inpaint_depth_control.ckpt --prompt '{prompt}' --seed {seed}"
     try:
         result = subprocess.run(command, shell=True, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
@@ -17,7 +18,7 @@ def generate(image, prompt, seed):
         for item in contents:
             print(item)
         
-        return "Done"
+        return f"output/{image_name}_512_0.jpg"
     except subprocess.CalledProcessError as e:
         print("Error:", e.stderr)
         return None
@@ -29,8 +30,8 @@ with gr.Blocks() as demo:
           textbox = gr.Textbox(show_label=False, value="a person facing the camera, making a hand gesture, indoor")
           seed = gr.Slider(minimum=0, maximum=1000000, value=643534)
           button = gr.Button()
-      #output_image = gr.Image(show_label=False, type="filepath", interactive=False, height=512, width=512)
-      output_image = gr.Textbox()
+      output_image = gr.Image(show_label=False, type="filepath", interactive=False, height=512, width=512)
+      #output_image = gr.Textbox()
     button.click(fn=generate, inputs=[image, textbox, seed], outputs=[output_image])
 
 demo.queue().launch(inline=False, share=True, debug=True)