Update app.py

app.py

@@ -1,17 +1,10 @@
-from huggingface_hub import from_pretrained_fastai
 import gradio as gr
-from fastai.vision.all import *
 from icevision.all import *
 
-# repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
-# repo_id = "igmarco/raccoon-image-detection"
-
 model1 = models.torchvision.faster_rcnn.model(backbone=models.torchvision.faster_rcnn.backbones.resnet18_fpn, num_classes=len(class_map))
 state_dict = torch.load('fasterRCNN_resnet18_Raccoons.pth')
 model1.load_state_dict(state_dict)
 
-# model1 = from_pretrained_fastai(repo_id)
-
 def show_preds(input_image, display_label, display_bbox, detection_threshold):
 
     if detection_threshold==0: detection_threshold=0.5