diff --git a/app/service/super_resolution/service.py b/app/service/super_resolution/service.py index 9fb9cac..1cfd386 100644 --- a/app/service/super_resolution/service.py +++ b/app/service/super_resolution/service.py @@ -59,8 +59,7 @@ class SuperResolution: @RunTime def infer(self, inputs): return self.triton_client.async_infer( - # model_name=SR_MODEL_NAME, - model_name="super_resolution_size-2", + model_name=SR_MODEL_NAME, inputs=inputs, callback=self.callback )