Update app.py

app.py

@@ -89,9 +89,33 @@ iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 
 iface.launch()
 """
+state_info = gr.Textbox(label="State", show_label=False, max_lines=2).style(container=False)
+
+def update_state_info(old_state):
+  if state and state != old_state:
+    return gr.update(value=state)
+
+def update_state(new_state):
+  global state
+  state = new_state
+    
+def pipe_callback(step: int, timestep: int, latents: torch.FloatTensor):
+    update_state(f"{step}/{current_steps} steps")#\nTime left, sec: {timestep/100:.0f}")
+
+
 def image_mod(image):
-    output1=proc11("full body sexy girl")
-    output1.save("save001.png")
+    #output1=proc11("full body sexy girl")
+    output1 = proc11(
+      prompt="full body sexy girl",
+      negative_prompt = "bad anatomy",
+      num_images_per_prompt=1,
+      num_inference_steps = int(10),
+      guidance_scale = 25,
+      width = 512,
+      height = 800,
+      generator = 0,
+      callback=pipe_callback)
+    #output1.save("save001.png")
     return(output1)
     #return image.rotate(45)
 
@@ -110,6 +134,7 @@ demo = gr.Interface(
     ],
     
 )
+demo.load(update_state_info, inputs=state_info, outputs=state_info, every=0.5, show_progress=False)
 
 
 demo.launch()