Update app.py

app.py

@@ -1,4 +1,16 @@
 import gradio as gr
+from transformers import pipeline
 
-gr.Interface.load(
-             "flyswot/convnext-tiny-224_flyswot",src='huggingface',interpretation="shap").launch()
+pipe = pipeline("image-classification", model="flyswot/convnext-tiny-224_flyswot")
+
+
+def predict(image):
+  predictions = pipe(image)
+  return {pred['label']: pred['score'] for pred in predictions}
+
+iface = gr.Interface(
+  fn=predict, 
+  inputs=gr.inputs.Image(type='filepath'),
+  outputs='label', interpretation='shap',num_shap=3)
+
+iface.launch()