Update app.py

app.py

@@ -1,5 +1,6 @@
 from fastai.vision.models.unet import DynamicUnet
 from torchvision.models.resnet import resnet18
+from fastai.vision.models import resnet18
 from fastai.vision.learner import create_body
 import streamlit as st
 from PIL import Image
@@ -355,7 +356,7 @@ from fastai.vision.models.unet import DynamicUnet
 
 def build_res_unet(n_input=1, n_output=2, size=256):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    body = create_body(resnet18, pretrained=True, n_in=n_input, cut=-2)
+    body = create_body(resnet18(pretrained=True), n_in=n_input, cut=-2)
     net_G = DynamicUnet(body, n_output, (size, size)).to(device)
     return net_G