Hugging Face's logo

Spaces:
AndreasLH
/
Weakly-Supervised-3DOD
Running

App Files Community
AndreasLH commited on
Commit
b11fef5
·
1 Parent(s): b530ea8

specify torch version to mitigate errors

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. app.py +1 -0
  3. pre-requirements.txt +1 -1
README.md CHANGED
@@ -2,7 +2,7 @@
2
  title: Weakly-Supervised-3DOD
3
  emoji: ☀️
4
  colorFrom: indigo
5
- colorTo: blue
6
  sdk: docker
7
  pinned: false
8
  license: apache-2.0
 
2
  title: Weakly-Supervised-3DOD
3
  emoji: ☀️
4
  colorFrom: indigo
5
+ colorTo: yellow
6
  sdk: docker
7
  pinned: false
8
  license: apache-2.0
app.py CHANGED
@@ -45,6 +45,7 @@ def do_test(im, threshold, model_str):
45
  # have to do this small workaround to only load the models once
46
  global model_loaded
47
  global model, depth_model, ground_model, segmentor
 
48
  if model_loaded != model_str:
49
  model, depth_model, ground_model, segmentor = load_model_config(model_str)
50
  model_loaded = model_str
 
45
  # have to do this small workaround to only load the models once
46
  global model_loaded
47
  global model, depth_model, ground_model, segmentor
48
+ print(model_loaded)
49
  if model_loaded != model_str:
50
  model, depth_model, ground_model, segmentor = load_model_config(model_str)
51
  model_loaded = model_str
pre-requirements.txt CHANGED
@@ -3,7 +3,7 @@ iopath
3
  fvcore
4
 
5
  --extra-index-url https://download.pytorch.org/whl/cpu
6
- torch
7
  torchvision
8
  torchaudio
9
 
 
3
  fvcore
4
 
5
  --extra-index-url https://download.pytorch.org/whl/cpu
6
+ torch==2.1
7
  torchvision
8
  torchaudio
9