Model.py

from transformers import pipeline

classifier = pipeline("image-classification", "JJNeila/fish_classification")

class FishModel:
    def __init__(self):
        self.classifier = classifier
    
    def get_fish_species(self, image):
        return self.classifier(image)