Update app.py

app.py

@@ -10,7 +10,13 @@ except ImportError:
                            "torchvision==0.15.2+cpu", 
                            "-f", "https://download.pytorch.org/whl/torch_stable.html"])
 
+try:
+    import numpy as np
+except ImportError:
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "numpy<2"])
+
 import torch
+import numpy
 import torch.nn as nn
 import torchvision.transforms as transforms
 from PIL import Image