Increase num_steps in inference pipeline to 9, bug fix

.ipynb_checkpoints/app-checkpoint.py

@@ -24,7 +24,7 @@ def inference(prompt, model, n_images, seed):
     print(f"Inferencing '{prompt}' for {n_images} images.")
 
     for i in range(n_images):
-        result = pipe(prompt, generator=generators[i], num_inference_steps=5).images[0]
+        result = pipe(prompt, generator=generators[i], num_inference_steps=9).images[0]
         result_images.append(result)
 
         # Start with empty UI elements
@@ -33,8 +33,8 @@ def inference(prompt, model, n_images, seed):
 
         # Update the progress bar
         with progress_bar_ui.container():
-            value = ((i+1)/(len(dataset)))
-            progress_bar.progress(value, text=f"{i+1} out of {len(dataset)} images processed.")
+            value = ((i+1)/n_images)
+            progress_bar.progress(value, text=f"{i+1} out of {n_images} images processed.")
 
         # Update the image grid
         with image_grid_ui.container():

app.py

@@ -24,7 +24,7 @@ def inference(prompt, model, n_images, seed):
     print(f"Inferencing '{prompt}' for {n_images} images.")
 
     for i in range(n_images):
-        result = pipe(prompt, generator=generators[i], num_inference_steps=5).images[0]
+        result = pipe(prompt, generator=generators[i], num_inference_steps=9).images[0]
         result_images.append(result)
 
         # Start with empty UI elements
@@ -33,8 +33,8 @@ def inference(prompt, model, n_images, seed):
 
         # Update the progress bar
         with progress_bar_ui.container():
-            value = ((i+1)/(len(dataset)))
-            progress_bar.progress(value, text=f"{i+1} out of {len(dataset)} images processed.")
+            value = ((i+1)/n_images)
+            progress_bar.progress(value, text=f"{i+1} out of {n_images} images processed.")
 
         # Update the image grid
         with image_grid_ui.container():