Thomas Male
commited on
Update handler.py
Browse files- handler.py +4 -4
handler.py
CHANGED
|
@@ -69,7 +69,7 @@ class EndpointHandler():
|
|
| 69 |
|
| 70 |
inputs = data.pop("inputs", data)
|
| 71 |
|
| 72 |
-
|
| 73 |
sampler = PointCloudSampler(
|
| 74 |
device=device,
|
| 75 |
models=[base_model, upsampler_model],
|
|
@@ -78,13 +78,13 @@ class EndpointHandler():
|
|
| 78 |
aux_channels=['R', 'G', 'B'],
|
| 79 |
guidance_scale=[3.0, 3.0],
|
| 80 |
)
|
| 81 |
-
|
| 82 |
# Load an image to condition on.
|
| 83 |
image_data = base64.b64decode(image_data_encoded)
|
| 84 |
-
|
| 85 |
# Convert bytes to PIL Image
|
| 86 |
img = Image.open(BytesIO(image_data))
|
| 87 |
-
|
| 88 |
sampler = PointCloudSampler(
|
| 89 |
device=device,
|
| 90 |
models=[self.base_model,self.upsampler_model],
|
|
|
|
| 69 |
|
| 70 |
inputs = data.pop("inputs", data)
|
| 71 |
|
| 72 |
+
if use_image:
|
| 73 |
sampler = PointCloudSampler(
|
| 74 |
device=device,
|
| 75 |
models=[base_model, upsampler_model],
|
|
|
|
| 78 |
aux_channels=['R', 'G', 'B'],
|
| 79 |
guidance_scale=[3.0, 3.0],
|
| 80 |
)
|
| 81 |
+
|
| 82 |
# Load an image to condition on.
|
| 83 |
image_data = base64.b64decode(image_data_encoded)
|
| 84 |
+
|
| 85 |
# Convert bytes to PIL Image
|
| 86 |
img = Image.open(BytesIO(image_data))
|
| 87 |
+
else:
|
| 88 |
sampler = PointCloudSampler(
|
| 89 |
device=device,
|
| 90 |
models=[self.base_model,self.upsampler_model],
|