Upload folder using huggingface_hub

.ipynb_checkpoints/main-checkpoint.py

@@ -4,8 +4,10 @@ from transformers import pipeline
 pipeline = pipeline("image-classification", model="jhoppanne/SkinCancerClassifier_Plain-V0")
 
 def predict(input_img):
-    predictions = pipeline(input_img)
-    return predictions
+    pred = pipeline(input_img)
+    label = ['Benign','Indeterminate','Malignant']
+    answer = f'We predict that you have {label[pred['label']]} type of skin cancer,\n with confidence score of: {pred['score']*100:.2f}%'
+    return answer
 
 gradio_app = gr.Interface(
     predict,

main.py

@@ -4,8 +4,10 @@ from transformers import pipeline
 pipeline = pipeline("image-classification", model="jhoppanne/SkinCancerClassifier_Plain-V0")
 
 def predict(input_img):
-    predictions = pipeline(input_img)
-    return predictions
+    pred = pipeline(input_img)
+    label = ['Benign','Indeterminate','Malignant']
+    answer = f'We predict that you have {label[pred['label']]} type of skin cancer,\n with confidence score of: {pred['score']*100:.2f}%'
+    return answer
 
 gradio_app = gr.Interface(
     predict,