Update app.py

app.py

@@ -6,8 +6,6 @@ from fastbook import PILImage
 # Load the pretrained FastAI model
 learn = load_learner("model.pkl")
 
-
-
 def classify_image(image):
     # Convert Gradio input to PIL Image
     image_pil = Image.fromarray(image.astype('uint8'), 'RGB')
@@ -18,12 +16,12 @@ def classify_image(image):
     # Perform inference
     classification,_,probs = learn.predict(PILImage.create('temp_image.jpg'))
     
-    return f"Predicted class: {classification}, Probabilities: {probs:.2f}"
+    return f"This is a {classification}"
 
 # Define Gradio interface
-inputs = gr.inputs.Image(shape=(224, 224))
-outputs = gr.outputs.Textbox(label="Prediction")
-interface = gr.Interface(fn=classify_image, inputs=inputs, outputs=outputs, capture_session=True)
+inputs = gr.Image()
+label = gr.Textbox(label="Prediction")
+interface = gr.Interface(fn=classify_image, inputs=inputs, outputs=label)
 
 # Run the interface
 interface.launch(share=True)