Hugging Face's logo

Spaces:
panchadip
/
sentiment2-test
Sleeping

App Files Community
panchadip commited on
Commit
6a1cd8b
·
verified ·
1 Parent(s): e202639

Update models/load_model.py

Browse files
Files changed (1) hide show
  1. models/load_model.py +1 -0
models/load_model.py CHANGED
@@ -1,5 +1,6 @@
1
  import joblib
2
  import torch
 
3
 
4
  # Define a custom load function to force CPU loading
5
  def custom_torch_load(f, *args, **kwargs):
 
1
  import joblib
2
  import torch
3
+ from huggingface_hub import hf_hub_download
4
 
5
  # Define a custom load function to force CPU loading
6
  def custom_torch_load(f, *args, **kwargs):