yashvshetty commited on
Commit
2b19796
·
1 Parent(s): a58d6dd

Add trust_remote_code=True to MedASR pipeline

Browse files
Files changed (1) hide show
  1. backend/models/medasr.py +1 -0
backend/models/medasr.py CHANGED
@@ -82,6 +82,7 @@ class MedASRModel:
82
  model=self.settings.MEDASR_MODEL_ID,
83
  revision="2625be4f1377ac544b451c6938eaf955c19a9c38",
84
  device=device,
 
85
  )
86
  except Exception as exc:
87
  raise ModelExecutionError(f"Failed to load MedASR model: {exc}") from exc
 
82
  model=self.settings.MEDASR_MODEL_ID,
83
  revision="2625be4f1377ac544b451c6938eaf955c19a9c38",
84
  device=device,
85
+ trust_remote_code=True,
86
  )
87
  except Exception as exc:
88
  raise ModelExecutionError(f"Failed to load MedASR model: {exc}") from exc