Update app.py

app.py

@@ -1,6 +1,12 @@
 from huggingface_hub import from_pretrained_fastai
 import gradio as gr
 import numpy as np
+from fastai.basics import *
+from fastai.vision import models
+from fastai.vision.all import *
+from fastai.metrics import *
+from fastai.data.all import *
+from fastai.callback import *
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu") 
 model = torch.jit.load("hrnet.pth")