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,
|