Update app.py

app.py

@@ -61,7 +61,7 @@ def infer(prompt,
     with torch.inference_mode():
         # 1. Create a prompt to generate an initial image
         output = vlm_pipe(prompt=prompt)
-        json_prompt_generate = output.values["json_prompt"]
+        json_prompt = output.values["json_prompt"]
 
 
         image = pipe(prompt=json_prompt, 
@@ -73,7 +73,7 @@ def infer(prompt,
 
     
     
-    return image
+    return image, json_prompt
 
 css = """
 #col-container{
@@ -142,12 +142,12 @@ with gr.Blocks(css=css) as demo:
                 submit_btn = gr.Button("Generate")
         result = gr.Image(label="output")
 
-    prompt_in.change(
-        handle_json, 
-        inputs=prompt_in, 
-        outputs=prompt_in_json)
+    # prompt_in.change(
+    #     handle_json, 
+    #     inputs=prompt_in, 
+    #     outputs=prompt_in_json)
 
-    negative_prompt.change(handle_json, inputs=negative_prompt, outputs=negative_prompt_json)
+    # negative_prompt.change(handle_json, inputs=negative_prompt, outputs=negative_prompt_json)
     
     submit_btn.click(
         fn = infer,
@@ -162,7 +162,7 @@ with gr.Blocks(css=css) as demo:
             num_inference_steps,
         ],
         outputs = [
-            result
+            result, prompt_in_json
         ]
     )
 demo.queue().launch()