Update app.py

app.py

@@ -2,6 +2,7 @@ import torch
 import torch.nn.functional as F
 import torchvision.transforms as transforms
 from PIL import Image
+from huggingface_hub import hf_hub_download
 import gradio as gr
 
 model_path = hf_hub_download(repo_id="Ayamohamed/DiaClassification", filename="dia_none_classifier_full.pth")