update

app.py

@@ -45,8 +45,6 @@ def get_mask_from_rectangles(image, mask, height, width, drawing_mode='rect'):
             objects[col] = objects[col].astype("str")
 
         if len(objects) > 0:
-            # st.dataframe(objects)
-
             left_coords = objects.left.to_numpy()
             top_coords = objects.top.to_numpy()
             right_coords = left_coords + objects.width.to_numpy()
@@ -75,7 +73,6 @@ def get_mask(image, edit_method, height, width):
         viridis = mpl.colormaps.get_cmap('viridis').resampled(np.max(seg))
         seg_image = Image.fromarray(np.uint8(viridis(seg)*255))
 
-        # display image
         st.image(seg_image)
 
         # prompt user to select valid labels to edit
@@ -103,7 +100,6 @@ if __name__ == '__main__':
 
     # upload image
     filename = st.file_uploader("upload an image")
-    # filename = r"C:\Users\itber\Downloads\Fjord_Cycling.jpg"
 
     sf = st.text_input("Please enter resizing scale factor to downsize image (default = 2)", value="2")
     try: 
@@ -130,10 +126,16 @@ if __name__ == '__main__':
 
             # get inpainted images
             prompt = st.text_input("Please enter prompt for image inpainting", value="")
+            seed = st.text_input("(Optional) enter seed to change inpainting result (default=0)", value="0")
+            try:
+                seed = int(seed)
+            except:
+                st.write("Invalid seed! Defaultign to 0, please re-enter above to change it")
+                seed = 0
 
             st.write("Inpainting Images, patience is a virtue :)")
 
-            images = inpaint(image, mask, width, height, prompt=prompt, seed=0, guidance_scale=17.5, num_samples=3)
+            images = inpaint(image, mask, width, height, prompt=prompt, seed=seed, guidance_scale=17.5, num_samples=3)
 
             # display all images
             st.write("Original Image")

model.py

@@ -50,7 +50,7 @@ def inpaint(image, mask, W, H, prompt="", seed=0, guidance_scale=17.5, num_sampl
         Outputs:
             images - output images
         """
-    generator = torch.Generator(device="cuda").manual_seed(seed)
+    generator = torch.Generator(device=device).manual_seed(seed)
     images = pipe(
         prompt=prompt,
         image=image,