Update app.py
Browse files
    	
        app.py
    CHANGED
    
    | 
         @@ -82,9 +82,10 @@ def generate( 
     | 
|
| 82 | 
         | 
| 83 | 
         
             
                pipe.to(device)
         
     | 
| 84 | 
         
             
                seed = int(randomize_seed_fn(seed, randomize_seed))
         
     | 
| 85 | 
         
            -
             
     | 
| 86 | 
         
             
                if not use_negative_prompt:
         
     | 
| 87 | 
         
             
                    negative_prompt = ""  # type: ignore
         
     | 
| 
         | 
|
| 88 | 
         
             
                if model == "Alpha":
         
     | 
| 89 | 
         
             
                    images = pipe(
         
     | 
| 90 | 
         
             
                        prompt=prompt,
         
     | 
| 
         @@ -97,6 +98,7 @@ def generate( 
     | 
|
| 97 | 
         
             
                        output_type="pil",
         
     | 
| 98 | 
         
             
                    ).images
         
     | 
| 99 | 
         
             
                elif model == "Alpha inpaint":
         
     | 
| 
         | 
|
| 100 | 
         
             
                    blurred_mask = pipe_inpaint.mask_processor.blur(mask_image, blur_factor=blur_factor)
         
     | 
| 101 | 
         
             
                    images = pipe_inpaint(
         
     | 
| 102 | 
         
             
                        prompt=prompt,
         
     | 
| 
         | 
|
| 82 | 
         | 
| 83 | 
         
             
                pipe.to(device)
         
     | 
| 84 | 
         
             
                seed = int(randomize_seed_fn(seed, randomize_seed))
         
     | 
| 85 | 
         
            +
                
         
     | 
| 86 | 
         
             
                if not use_negative_prompt:
         
     | 
| 87 | 
         
             
                    negative_prompt = ""  # type: ignore
         
     | 
| 88 | 
         
            +
                images = None
         
     | 
| 89 | 
         
             
                if model == "Alpha":
         
     | 
| 90 | 
         
             
                    images = pipe(
         
     | 
| 91 | 
         
             
                        prompt=prompt,
         
     | 
| 
         | 
|
| 98 | 
         
             
                        output_type="pil",
         
     | 
| 99 | 
         
             
                    ).images
         
     | 
| 100 | 
         
             
                elif model == "Alpha inpaint":
         
     | 
| 101 | 
         
            +
                    print("Inpainting...")
         
     | 
| 102 | 
         
             
                    blurred_mask = pipe_inpaint.mask_processor.blur(mask_image, blur_factor=blur_factor)
         
     | 
| 103 | 
         
             
                    images = pipe_inpaint(
         
     | 
| 104 | 
         
             
                        prompt=prompt,
         
     |