Update app.py

app.py

@@ -1,12 +1,16 @@
 import gradio as gr
 
-def image_classifier(input):
-    return {'normal': 0.3, 'bullying': 0.7}
-   
-demo = gr.Interface.load(
-    "models/datasciencemmw/pickonai-best",
-    inputs="image",
-    output="label",
-    title="PickOnAI Demo")
-demo.launch()
+from transformers import pipeline
 
+pipe = pipeline("image-classification", model="datasciencemmw/pickonai-best")
+
+def predict(image):
+  return pipe(image)[0]["image_classification"]
+
+iface = gr.Interface(
+  fn=predict, 
+  inputs='image',
+  outputs='label',
+)
+
+iface.launch()