Spaces:
Runtime error
Runtime error
File size: 1,371 Bytes
e0b1654 8a82feb 927cd84 0e20c0a 3ca4172 0e20c0a eeb12be 0e20c0a 3ca4172 ae8be33 0e20c0a c023b45 ac391cc 68998a5 e0b1654 c023b45 6e8dace |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
import gradio as gr
from image_dataset import ImageDataset
from image_wgan import ImageWgan
import os
from os.path import exists
def init():
generated_samples_folder = "results64"
discriminator_saved_model = "discriminator64.model"
generator_saved_model = "generator64.model"
latent_space = 100
image_wgan = ImageWgan(
image_shape = (4,64,64),
latent_space_dimension=latent_space,
generator_saved_model=generator_saved_model if exists(generator_saved_model) else None,
discriminator_saved_model=discriminator_saved_model if exists(discriminator_saved_model) else None
)
image_wgan.generate(
sample_folder=generated_samples_folder
)
crop()
def crop():
from PIL import Image
import generator
res = 64
if res != 0:
results = "results64/generated.png"
img = Image.open(results)
width,height = img.size
top = 2
bottom = 2
for i in range(4):
left = (res+2)*i +2
right = width-(res+2)*i
imgcrop = img.crop((left,top,left+res,res+2))
imgcrop.save("results64/"+str(i)+".png")
init()
def gen():
init()
crop()
img = Image.open("results64/0.png")
print(img)
return "hi"
iface = gr.Interface(fn=gen, inputs=None, outputs="text")
iface.launch(debug = True) |