Update app.py

app.py

@@ -18,6 +18,7 @@ from PIL import Image
 def get_model(args,arch,load_from,arch_path):
     if load_from == 'timm':
         model = create_model(arch,pretrained = True).to(args.PARAMETERS.device)
+        print("Load model timm")
     elif load_from == 'torchvision':
         if arch == 'resnet50':
             model = torchvision.models.resnet50(pretrained=False)