Update app.py

app.py

@@ -1,8 +1,7 @@
 # prompt: gradio image 分类
 import fastai
 from fastai.vision import *
-from fastai.vision.learner import load_learner
-from PIL import Image 
+from fastai.vision.all import load_learner,PILImage
 import gradio as gr
 
 # Load the model
@@ -10,14 +9,13 @@ model = load_learner("model.pkl")
 
 # Define an image classification function
 def classify_image(image):
-    # Preprocess the image
-    image = Image.create(image).resize((192, 192))
+    img = PILImage.create(image)
 
     # Make a prediction
-    prediction = model.predict(image)[0]
-
-    # Return the prediction
-    return {prediction: 1.0}
+    pred_class, pred_idx, probs = model.predict(img)
+    
+    # Return the prediction as a dictionary
+    return {model.dls.vocab[i]: float(probs[i]) for i in range(len(probs))}
 
 # Create the Gradio interface
 image_input = gr.Image()