Alesteba commited on
Commit
ab9e38d
·
1 Parent(s): b414639

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -7
app.py CHANGED
@@ -1,10 +1,9 @@
1
- # imports
2
-
3
- import os
4
-
5
- # from icevision.all import *
6
- import torch
7
- from torchvision import *
8
 
9
  model_2 = models.torchvision.retinanet.model(
10
  backbone=models.torchvision.retinanet.backbones.resnext50_32x4d_fpn (pretrained=True),
 
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
+ from icevision.models.checkpoint import *
6
+ import PIL
 
7
 
8
  model_2 = models.torchvision.retinanet.model(
9
  backbone=models.torchvision.retinanet.backbones.resnext50_32x4d_fpn (pretrained=True),