jph00/pets · Update requirements and subsequent model.pkl

Jun 13, 2023

•

edited Jun 13, 2023

Saw that the space is having issues running.
I cloned locally and had to change the environment to train and run the gradio demo on a computer with an Nvidia GPU.
Did not need to change any of the code in the notebooks.

Update requirements.txt and subsequent model.pkl25000727

Remove jlab as dep for hface deployment8454c1e7

evylz changed pull request status to open Jun 13, 2023

evylz

Jun 13, 2023

This should resolve the Runtime Error

Traceback (most recent call last):
  File "/home/user/app/app.py", line 11, in <module>
    learn = load_learner('model.pkl')
  File "/home/user/.local/lib/python3.10/site-packages/fastai/learner.py", line 446, in load_learner
    try: res = torch.load(fname, map_location=map_loc, pickle_module=pickle_module)
  File "/home/user/.local/lib/python3.10/site-packages/torch/serialization.py", line 712, in load
    return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
  File "/home/user/.local/lib/python3.10/site-packages/torch/serialization.py", line 1046, in _load
    result = unpickler.load()
  File "/home/user/.local/lib/python3.10/site-packages/torch/serialization.py", line 1039, in find_class
    return super().find_class(mod_name, name)
ModuleNotFoundError: No module named 'timm.models.layers.mlp'

jph00 changed pull request status to merged Jun 13, 2023

jph00

Owner Jun 13, 2023

Many thanks!