lfolle commited on
Commit
86ab458
1 Parent(s): 3110826

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -8,8 +8,7 @@ from huggingface_hub import hf_hub_download
8
 
9
  file_path = hf_hub_download("lfolle/DeepNAPSIModel", "dummy_model.pt",
10
  use_auth_token=os.environ['DeepNAPSIModel'])
11
- print(file_path)
12
- # model = torch.load(file_path)
13
 
14
  def predict(data):
15
  return model(data)
 
8
 
9
  file_path = hf_hub_download("lfolle/DeepNAPSIModel", "dummy_model.pt",
10
  use_auth_token=os.environ['DeepNAPSIModel'])
11
+ model = torch.load(file_path)
 
12
 
13
  def predict(data):
14
  return model(data)