Update app.py

app.py

@@ -5,6 +5,7 @@ import gradio as gr
 
 with open("birds.csv", "r") as r:
   birds = list(csv.reader(r, delimiter=","))
+  birds = birds[0]
 
 config = {"num_channels": 3, "hidden_units": 256, "num_classes": 525, "labels": birds}
 
@@ -20,7 +21,7 @@ def predice(imagen):
     transforms.Resize(size=(224, 224)),
     ])
     
-    target_image = io.read_image(imagen).type(torch.float32)
+    target_image = io.read_image(str(imagen)).type(torch.float32)
     target_image = target_image / 255.
     target_image = usr_img_transform(target_image)
     
@@ -29,12 +30,12 @@ def predice(imagen):
       target_image = target_image.unsqueeze(dim=0)
       target_image_pred = model(target_image)
     target_image_pred_label = torch.argmax(target_image_pred, dim=1)
-    label = config["labels"][target_image_pred_label]
-    return label
+    label = config["labels"][target_image_pred_label.item()]
+    return str(label)
 
 gr.Interface(
     predice,
-    inputs=gr.Image(label="gimme da bird"),
+    inputs=gr.Image(label="gimme da bird", type="filepath"),
     outputs="label",
     title=titulo,
     description=desc,