Update app.py

app.py

@@ -160,8 +160,8 @@ def predict(radio, dict, word_mask, prompt=""):
             mask = Image.fromarray(np.uint8(bw_image)).convert('RGB')
             os.remove(filename)
         with autocast(device): #"cuda"
-        images = pipe(prompt = prompt, init_image=init_image, mask_image=mask, strength=0.8)["sample"]
-    return images[0]
+            images = pipe(prompt = prompt, init_image=init_image, mask_image=mask, strength=0.8)["sample"]
+        return images[0]
 
 # examples = [[dict(image="init_image.png", mask="mask_image.png"), "A panda sitting on a bench"]]
 css = '''