no old size, yes inference steps, warm up on two sizes

app.py

@@ -66,7 +66,6 @@ def get_seg_mask(img):
 
 def app(img, prompt, num_inference_steps, seed, inpaint_size):
     start_time = datetime.now().timestamp()
-    old_size = Image.fromarray(img).size
     img = np.array(Image.fromarray(img).resize(seg_working_size))
     mask = get_seg_mask(img)
     # mask.save("mask.jpg")
@@ -88,15 +87,21 @@ def app(img, prompt, num_inference_steps, seed, inpaint_size):
     # replace spaces with newlines after many words to line break prompt
     prompt = " ".join([prompt.split(" ")[i] if (i+1) % 5 else prompt.split(" ")[i] + "\n" for i in range(len(prompt.split(" ")))])
 
-    draw.text((10, 50), f"Old size: {old_size}\nTotal duration: {int(1000 * (end_time - start_time))}ms\nSegmentation {int(1000 * (mask_time - start_time))}ms / inpainting {int(1000 * (end_time - mask_time))} \n<{prompt}>", fill=(0, 255, 0))
+    draw.text((10, 50), "\n".join([
+            f"Total duration: {int(1000 * (end_time - start_time))}ms",
+            f"Inference steps: {num_inference_steps}",
+            f"Segmentation {int(1000 * (mask_time - start_time))}ms / inpainting {int(1000 * (end_time - mask_time))}",
+            f"<{prompt}>"
+        ]), fill=(0, 255, 0))
     #overlay_img.save("overlay_with_text.jpg")
     return overlay_img
 
 # warmup, for compiling and then for timing
 
-for i in range(2):
-    for j in tqdm(range(3 ** i)):
-        app(np.array(Image.fromarray(np.zeros((1024,1024,3), dtype=np.uint8))), default_inpainting_prompt, 4, 42, 512).save("zeros_inpainting_oneshot.jpg")
+for size in [384,512]:
+    for i in range(2):
+        for j in tqdm(range(3 ** i)):
+            app(np.array(Image.fromarray(np.zeros((1024,1024,3), dtype=np.uint8))), default_inpainting_prompt, 4, 42, size).save("zeros_inpainting_oneshot.jpg")
 
 #ideally:
 #iface = gr.Interface(app, gr.Image(sources=["webcam"], streaming=True), "image", live=True)
@@ -108,4 +113,4 @@ iface = gr.Interface(app, [
         gr.Number(value=512, maximum=seg_model_img_size,)
     ],
     "image")
-iface.launch()
+iface.launch(share=True)