Update app.py

app.py

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