Update app.py

app.py

@@ -1,7 +1,26 @@
+# prompt: gradio image 分类
+import fastai
+from fastai.vision import *
 import gradio as gr
 
-def greet(name):
-    return "Hello wdq -- " + name + "!"
+# Load the model
+model = load_learner("model.pkl")
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+# Define an image classification function
+def classify_image(image):
+    # Preprocess the image
+    image = PILImage.create(image).resize((192, 192))
+
+    # Make a prediction
+    prediction = model.predict(image)[0]
+
+    # Return the prediction
+    return {prediction: 1.0}
+
+# Create the Gradio interface
+image_input = gr.Image()
+label_output = gr.Label(num_top_classes=3)
+interface = gr.Interface(fn=classify_image, inputs=image_input, outputs=label_output)
+
+# Launch the interface
+interface.launch()