Update app.py

app.py

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