Spaces:
Runtime error
Runtime error
Commit
·
362659d
1
Parent(s):
c99ccb0
Update app.py
Browse files
app.py
CHANGED
|
@@ -15,6 +15,7 @@ transform = Compose(
|
|
| 15 |
model = GeneratorUNet.from_pretrained('huggan/pix2pix-edge2shoes')
|
| 16 |
|
| 17 |
def predict_fn(img):
|
|
|
|
| 18 |
inp = transform(img).unsqueeze(0)
|
| 19 |
out = model(inp)
|
| 20 |
save_image(out, 'out.png', normalize=True)
|
|
|
|
| 15 |
model = GeneratorUNet.from_pretrained('huggan/pix2pix-edge2shoes')
|
| 16 |
|
| 17 |
def predict_fn(img):
|
| 18 |
+
img = Image.fromarray(img.astype('uint8'), 'RGB')
|
| 19 |
inp = transform(img).unsqueeze(0)
|
| 20 |
out = model(inp)
|
| 21 |
save_image(out, 'out.png', normalize=True)
|