Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -154,7 +154,7 @@ def infer(use_custom_model, model_name, weight_name, custom_lora_weight, image_i
|
|
154 |
image = np.array(image)
|
155 |
image = HWC3(image)
|
156 |
#image = resize_image(image, resolution=image_resolution)
|
157 |
-
image =
|
158 |
|
159 |
if use_custom_model:
|
160 |
|
|
|
154 |
image = np.array(image)
|
155 |
image = HWC3(image)
|
156 |
#image = resize_image(image, resolution=image_resolution)
|
157 |
+
image = Image.fromarray(image)
|
158 |
|
159 |
if use_custom_model:
|
160 |
|