Spaces:
Runtime error
Runtime error
Update model.py
Browse files
model.py
CHANGED
|
@@ -177,7 +177,7 @@ class Model:
|
|
| 177 |
127.5).cpu().numpy().clip(0, 255).astype(np.uint8)
|
| 178 |
|
| 179 |
results = [x_samples[i] for i in range(num_samples)]
|
| 180 |
-
return [
|
| 181 |
|
| 182 |
@torch.inference_mode()
|
| 183 |
def process_hough(self, input_image, prompt, a_prompt, n_prompt,
|
|
|
|
| 177 |
127.5).cpu().numpy().clip(0, 255).astype(np.uint8)
|
| 178 |
|
| 179 |
results = [x_samples[i] for i in range(num_samples)]
|
| 180 |
+
return [input_image] + results
|
| 181 |
|
| 182 |
@torch.inference_mode()
|
| 183 |
def process_hough(self, input_image, prompt, a_prompt, n_prompt,
|