Stable-X commited on
Commit
10cae83
1 Parent(s): d3b62a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -23,8 +23,7 @@ class Examples(gr.helpers.Examples):
23
 
24
  def load_predictor():
25
  """Load model predictor using torch.hub"""
26
- predictor = torch.hub.load("hugoycj/StableNormal", "StableNormal", trust_repo=True,
27
- local_cache_dir='./weights')
28
  return predictor
29
 
30
  def process_image(
 
23
 
24
  def load_predictor():
25
  """Load model predictor using torch.hub"""
26
+ predictor = torch.hub.load("hugoycj/StableNormal", "StableNormal", trust_repo=True)
 
27
  return predictor
28
 
29
  def process_image(