juancopi81 commited on
Commit
0627c0d
1 Parent(s): a3ff253

Add correct path to checkpoint

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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'/content/checkpoints/{model}/'
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