Adityak204 commited on
Commit
6fae1ea
·
1 Parent(s): ea55197
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +2 -1
app.py CHANGED
@@ -6,6 +6,7 @@ import json
6
  from pathlib import Path
7
  import os
8
  from huggingface_hub import hf_hub_download
 
9
 
10
 
11
  class ModelPredictor:
 
6
  from pathlib import Path
7
  import os
8
  from huggingface_hub import hf_hub_download
9
+ import numpy as np
10
 
11
 
12
  class ModelPredictor:
requirements.txt CHANGED
@@ -4,4 +4,5 @@ torchvision
4
  pytorch-lightning
5
  loguru
6
  Pillow
7
- huggingface_hub
 
 
4
  pytorch-lightning
5
  loguru
6
  Pillow
7
+ huggingface_hub
8
+ numpy