Update app.py

app.py

@@ -1,6 +1,11 @@
 import gradio as gr
 from transformers import pipeline
 
+def predict_image(img):
+  img_4d=img.reshape(-1,180,180,3)
+  prediction=model.predict(img_4d)[0]
+  return {class_names[i]: float(prediction[i]) for i in range(4)}
+
 pipe = pipeline('image-classification')
 
 image = gr.inputs.Image(shape=(180,180))