Update app.py

app.py

@@ -139,9 +139,7 @@ class PatchDiscriminator(nn.Module):
         # activation for the last layer of the model
         self.model = nn.Sequential(*model)
 
-    # when needing to make some repeatitive blocks of layers,
     def get_layers(self, ni, nf, k=4, s=2, p=1, norm=True, act=True):
-        # it's always helpful to make a separate method for that purpose
         layers = [nn.Conv2d(ni, nf, k, s, p, bias=not norm)]
         if norm:
             layers += [nn.BatchNorm2d(nf)]
@@ -350,6 +348,9 @@ def log_results(loss_meter_dict):
 
 
 # pip install fastai==2.4
+from fastai.vision.learner import create_body
+from torchvision.models.resnet import resnet18
+from fastai.vision.models.unet import DynamicUnet
 
 
 def build_res_unet(n_input=1, n_output=2, size=256):