Update app.py

app.py

@@ -13,9 +13,9 @@ classifier = VisionClassifierInference(
 
 def classify_image(image_file):
     """Classify an image using a pre-trained ViT model."""
-    label = classifier.predict(img_path=image_file.name)
+    label = classifier.predict(img=Image.open(image_file.name))
     # Add a confidence score to the output
-    confidence = classifier.predict_proba(img_path=image_file.name)[0][label]
+    confidence = classifier.predict_proba(img=Image.open(image_file.name))[0][label]
 
     # Get the PIL Image object for the uploaded image
     image = Image.open(image_file)
@@ -33,6 +33,7 @@ def classify_image(image_file):
     return output_image, f"Predicted class: {label} (confidence: {confidence:.2f})"
 
 
+
 iface = gr.Interface(
     fn=classify_image,
     inputs=gr.inputs.Image(type="filepath", label="Upload an image"),