Spaces:

ccareaga
/

Intrinsic

Running on Zero

ccareaga commited on Nov 22, 2023

Commit

6a18cf5

1 Parent(s): 4924b15

make sure model runs on cpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,12 +6,13 @@ from intrinsic.pipeline import run_pipeline
 from intrinsic.model_util import load_models
 global intrinsic_model
-intrinsic_model = load_models('paper_weights')
 def decompose(img):
     result = run_pipeline(
         img,
-        intrinsic_model
     )
     return view(uninvert(result['inv_shd'])), "Completed"

 from intrinsic.model_util import load_models
 global intrinsic_model
+intrinsic_model = load_models('paper_weights', device='cpu')
 def decompose(img):
     result = run_pipeline(
         img,
+        intrinsic_model,
+        device='cpu'
     )
     return view(uninvert(result['inv_shd'])), "Completed"