Spaces:
Build error
Build error
juancopi81
commited on
Commit
·
0627c0d
1
Parent(s):
a3ff253
Add correct path to checkpoint
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ os.putenv("QT_QPA_PLATFORM", "offscreen")
|
|
25 |
os.putenv("XDG_RUNTIME_DIR", environment.Environment().getRootTempDir())
|
26 |
|
27 |
def load_model(model=str):
|
28 |
-
checkpoint_path = f
|
29 |
# Start inference model
|
30 |
inference_model = InferenceModel(checkpoint_path, model)
|
31 |
return inference_model
|
|
|
25 |
os.putenv("XDG_RUNTIME_DIR", environment.Environment().getRootTempDir())
|
26 |
|
27 |
def load_model(model=str):
|
28 |
+
checkpoint_path = f"/home/user/app/checkpoints/{model}/"
|
29 |
# Start inference model
|
30 |
inference_model = InferenceModel(checkpoint_path, model)
|
31 |
return inference_model
|