I am working on building a CNN architecture model. After training it, I've uploaded it to Huggingface to make it deployable.

But I cannot use it as an interface because of this error

 ---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
File c:\Users\HP\OneDrive\Desktop\fine_tuning\.venv\Lib\site-packages\torch\serialization.py:851, in _check_seekable(f)
    850 try:
--> 851     f.seek(f.tell())
    852     return True

AttributeError: 'dict' object has no attribute 'seek'

During handling of the above exception, another exception occurred:

AttributeError                            Traceback (most recent call last)
Cell In[51], line 23
     21 # Load the trained model
     22 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
---> 23 model = AutoModelForImageClassification.from_pretrained("shahad-alh/Arabichar_Model-V1", trust_remote_code=True,
     24     id2label=id2label, label2id=label2id
     25 )
     26 model.to(device)
     27 model.eval()  # Set model to evaluation mode

File c:\Users\HP\OneDrive\Desktop\fine_tuning\.venv\Lib\site-packages\transformers\models\auto\auto_factory.py:568, in _BaseAutoModelClass.from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs)
    566     cls.register(config.__class__, model_class, exist_ok=True)
    567     model_class = add_generation_mixin_to_remote_model(model_class)
--> 568     return model_class.from_pretrained(
...
    846         )
--> 847         raise type(e)(msg)
    848 raise e

AttributeError: 'dict' object has no attribute 'seek'. You can only torch.load from a file that is seekable. Please pre-load the data into a buffer like io.BytesIO and try to load from it instead.

I believe that it's all because of the load_state_dict function

def load_state_dict(self, model_name,strict = False, assign = False): 
        self.model.load_state_dict(torch.load(model_name))

I'm happy to share more information if needed, hope someone knows what the problem is and how to solve it correctly.