Update app.py

app.py

@@ -3,7 +3,6 @@ from icevision.all import *
 import PIL
 import gradio as gr
 
-
 # Cargamos el learner
 learn = load_learner('export.pkl')
 
@@ -15,7 +14,7 @@ labels = learn.dls.vocab
 def predict(img):
     class_map = ['Kangaroo']
     model = models.torchvision.faster_rcnn.model(backbone=models.torchvision.faster_rcnn.backbones.resnet18_fpn, num_classes=len(class_map))
-state_dict = torch.load('model.pth')
+    state_dict = torch.load('model.pth')
     model.load_state_dict(state_dict)
     
     infer_tfms = tfms.A.Adapter([*tfms.A.resize_and_pad(size),tfms.A.Normalize()])