[Python] Cannot load FastAI .pkl model locally due to missing AMPMode and module version issues

history of issue :- I trained an efficientnet model with fastai on google colab and exported it to drive and than imported it to my local environment to make an gradio interface for that but upon running that I got an error of missing lzma model, than brew installed xz and updated python latest version to build with xz but upon retrying it showed error and asked to downgrade the fastai and fast core and also numpy to match the imported .pkl model. so for full security I checked the google colab python version and pyenv installed the same version (3.11.13) and set it local to my project folder and also removed old venv and made new one and also installed the required modules with required versions, than I got this error :-

"ImportError: cannot import name 'AMPMode' from 'fastai.callback.fp16' (/Users/nishchay/Documents/organic/orven/lib/python3.11/site-packages/fastai/callback/fp16.py)"

than I added this to my code but still getting the issue "from fastai.callback.fp16 import AMPMode" but still got the same issue

this is my full code :-

from fastai.callback.fp16 import AMPMode
import gradio as gr
from fastai.vision.all import *

# Load the model
try:
learner = load_learner("plant_diseases_healthy_model.pkl")
except FileNotFoundError:
print("Model file not found. Please ensure 'plant_diseases_healthy_model.pkl' is in the correct directory.")
exit()

# Define the prediction function
def classify_image(img):
preds, idx, probs = learner.predict(img)
return {learner.dls.vocab: float(probs) for i in range(len(probs))}

# Create the Gradio interface
interface = gr.Interface(
fn=classify_image,
inputs=gr.Image(type="pil", label="Upload Plant Image"),
outputs=gr.Label(num_top_classes=3, label="Predicted Disease"),
title="Plant Disease Classifier",
description="Upload an image of a plant leaf to identify potential diseases."
)

# Launch the interface
interface.launch()```

Continue reading...