Update app.py

app.py

@@ -3,9 +3,10 @@ import torch
 from PIL import Image
 import io
 import os
-import timm
 import torch.nn as nn
-
+import timm
+from torch import nn
+from torchvision import transforms
 # Create the 'model' directory if it doesn't exist (for saving the model later)
 if not os.path.exists('model'):
     os.makedirs('model')