inference.py

from roboflow import Roboflow
from PIL import Image
import numpy as np

# Load Roboflow model
rf = Roboflow(api_key="rSfidObyvZ1BYV7yOQrN")
project = rf.workspace("naveen-kumar-9emxl").project("my-first-project-icheu")
model = project.version(1).model

def predict(image):
    image = np.array(image)
    prediction = model.predict(image).plot()
    return Image.fromarray(prediction)