LogicGoInfotechSpaces commited on
Commit
d58eb50
·
1 Parent(s): bdef219

Point inference requests to new HF router endpoint

Browse files
Files changed (1) hide show
  1. app/colorize_model.py +1 -1
app/colorize_model.py CHANGED
@@ -65,7 +65,7 @@ class ColorizeModel:
65
 
66
  def __init__(self, model_id: str | None = None) -> None:
67
  self.model_id = model_id or settings.MODEL_ID
68
- self.api_url = f"https://api-inference.huggingface.co/models/{self.model_id}"
69
 
70
  self.api_token = (
71
  os.getenv("HUGGINGFACE_API_TOKEN")
 
65
 
66
  def __init__(self, model_id: str | None = None) -> None:
67
  self.model_id = model_id or settings.MODEL_ID
68
+ self.api_url = f"https://router.huggingface.co/hf-inference/models/{self.model_id}"
69
 
70
  self.api_token = (
71
  os.getenv("HUGGINGFACE_API_TOKEN")