Update app.py

app.py

@@ -26,6 +26,7 @@ def hf_inference(prompt, negative, model, steps, sampler, guidance, width, heigh
         images=[]
         time.sleep(1)
         progress(0.05)
+        progress(025, desc="Generaing")
         image_bytes = query(payload={
             "inputs": f"{prompt}",
             "parameters": {
@@ -36,8 +37,10 @@ def hf_inference(prompt, negative, model, steps, sampler, guidance, width, heigh
                 "seed": seed,
             },
         }, model=model)
+        progress(75, desc="Opening image")
         image = Image.open(io.BytesIO(image_bytes))
         images.append(image)
+        progress(99, desc="Sending image")
         return images
     except PIL.UnidentifiedImageError:
         gr.Warning("This model is not loaded now. Try others models.")
@@ -198,6 +201,9 @@ for model_name in model_list:
 
 
 def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
+    progress(0, desc="Starting")
+    time.sleep(2.5)
+    progress(25, desc="Generating")
     result = prodia_client.generate({
         "prompt": prompt,
         "negative_prompt": negative_prompt,
@@ -209,13 +215,18 @@ def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, he
         "height": height,
         "seed": seed
     })
-
+    progress(75, desc="Opening image")
     job = prodia_client.wait(result)
-
+    progress(99, desc="Sending image")
     return job["imageUrl"]
 
 
 def img2img(input_image, denoising, prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
+    progress(0, desc="Starting")
+    time.sleep(1.5)
+    progress(10, desc="Uploading input image")
+    time.sleep(1.5)
+    progress(25, desc="Generating")
     result = prodia_client.transform({
         "imageData": image_to_base64(input_image),
         "denoising_strength": denoising,
@@ -229,9 +240,10 @@ def img2img(input_image, denoising, prompt, negative_prompt, model, steps, sampl
         "height": height,
         "seed": seed
     })
+    progress(75, desc="Opening image")
 
     job = prodia_client.wait(result)
-
+    progress(99, desc="Sending image")
     return job["imageUrl"]