Update app.py

app.py

@@ -35,7 +35,7 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=20, s
     
     output = pipe(prompt = prompt, negative_prompt=negative_prompt, image=init_image, mask_image=mask, guidance_scale=guidance_scale, num_inference_steps=int(steps), strength=strength, clip_skip=1)
     output_image = output.images[0]  # Assuming the output dictionary has an 'images' key
-    output_image = F.interpolate(output_image.unsqueeze(0), size=(1024,832))[0]
+    output_image = torch.nn.functional.interpolate(output_image.unsqueeze(0), size=(1024,832))[0]
     return output.images, gr.update(visible=True)