Update app.py

app.py

@@ -4,10 +4,8 @@ import skimage
 
 learn = load_learner('model.pkl')
 
-# Original labels from the model
 labels = learn.dls.vocab
 
-# Custom mapping for labels
 custom_labels = {
     "Other": "Nimeshindwa kutambua picha",
     "lateblt": "Ukungu wa mwisho wa viazi",
@@ -18,13 +16,11 @@ custom_labels = {
 def predict(img):
     img = PILImage.create(img)
     pred, pred_idx, probs = learn.predict(img)
-    # Map the model's labels to the custom labels
     return {custom_labels[labels[i]]: float(probs[i]) for i in range(len(labels))}
 
-# Other parameters
 examples = ['image.jpg']
 interpretation = 'default'
 enable_queue = True
 
-# Launch Gradio interface
-gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label(num_top_classes=3), examples=examples, interpretation=interpretation, enable_queue=enable_queue).launch()
+# Updated component usage
+gr.Interface(fn=predict, inputs=gr.Image(shape=(512, 512)), outputs=gr.Label(num_top_classes=3), examples=examples, interpretation=interpretation, enable_queue=enable_queue).launch()