Update app.py

app.py

@@ -29,7 +29,7 @@ model.to(device=args.get("device"))
 torch.set_grad_enabled(False)
 
 
-def predict(image)->dict:
+def predict(image: str)->dict:
     global model, args, normtransform
     
     prediction_tensor = torch.zeros([1, len(args.get("dxlabels"))]).to(device=args.get("device"))
@@ -40,8 +40,8 @@ def predict(image)->dict:
     aug_combos = [x for x in itertools.product(target_sizes, hflips, rotations, crops)]
     
     # Load image
-    # img = Image.open(image)
-    img = image.convert('RGB')
+    img = Image.open(image)
+    img = img.convert('RGB')
 
     # Predict with Test-time augmentation
     for (target_size, hflip, rotation, crop) in aug_combos:
@@ -73,7 +73,7 @@ If you have a skin change in question, seek contact to your physician.'''
 
 gr.Interface(
     predict,
-    inputs=gr.Image(label="Upload a dermatoscopic image", type="pil"),
+    inputs=gr.Image(label="Upload a dermatoscopic image", type="filepath"),
     outputs=gr.Label(num_top_classes=len(args.get("dxlabels"))),
     title="Dermatoscopic classification",
     description=description,