Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
-
from huggingface_hub import from_pretrained_fastai
|
2 |
import gradio as gr
|
3 |
-
from fastai.vision.all import *
|
|
|
4 |
|
5 |
class_map = {'background': 0, 'kangaroo': 1}
|
6 |
model = models.torchvision.faster_rcnn.model(backbone=models.torchvision.faster_rcnn.backbones.resnet50_fpn,
|
|
|
1 |
+
# from huggingface_hub import from_pretrained_fastai
|
2 |
import gradio as gr
|
3 |
+
# from fastai.vision.all import *
|
4 |
+
from icevision.all import *
|
5 |
|
6 |
class_map = {'background': 0, 'kangaroo': 1}
|
7 |
model = models.torchvision.faster_rcnn.model(backbone=models.torchvision.faster_rcnn.backbones.resnet50_fpn,
|