Update app.py

app.py

@@ -74,8 +74,7 @@ def process_image(
     finally:
         os.remove(image_path)
 
-    return generated_image_url, result[1]
-
+    return gr.update(value=generated_image_url), result[1]
 # UI CSS
 css = """
 #col-container {