Update app.py

app.py

@@ -9,6 +9,11 @@ from fastai.callback import *
 import PIL
 import torchvision.transforms as transforms
 
+from huggingface_hub import hf_hub_download
+hf_hub_download(repo_id="Alesteba/deep_model_03", filename="unet.pth")
+
+# load model
+
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu") 
 model = torch.jit.load("pract3.pth")
 model = model.cpu()