Spaces:
Runtime error
Runtime error
Commit
·
53b7a6c
1
Parent(s):
91cec6b
Update app.py
Browse files
app.py
CHANGED
@@ -16,8 +16,6 @@ transform = Compose(
|
|
16 |
model = GeneratorUNet.from_pretrained('huggan/pix2pix-uavid-15')
|
17 |
|
18 |
def predict_fn(img):
|
19 |
-
img = Image.fromarray(np.array(img)[:, ::-1, :], "RGB")
|
20 |
-
img = img.convert("RGB")
|
21 |
inp = transform(img).unsqueeze(0)
|
22 |
out = model(inp)
|
23 |
save_image(out, 'out.png', normalize=True)
|
|
|
16 |
model = GeneratorUNet.from_pretrained('huggan/pix2pix-uavid-15')
|
17 |
|
18 |
def predict_fn(img):
|
|
|
|
|
19 |
inp = transform(img).unsqueeze(0)
|
20 |
out = model(inp)
|
21 |
save_image(out, 'out.png', normalize=True)
|