Commit
·
d58eb50
1
Parent(s):
bdef219
Point inference requests to new HF router endpoint
Browse files- 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://
|
| 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")
|