Spaces:
Runtime error
Runtime error
File size: 1,083 Bytes
5eeecb1 86d5bb4 5eeecb1 86d5bb4 5eeecb1 86d5bb4 5eeecb1 86d5bb4 5eeecb1 |
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 |
import gradio as gr
from huggingface_hub import PyTorchModelHubMixin
import torch
import matplotlib.pyplot as plt
import torchvision
from networks_fastgan import Generator
def image_generation(model, number_of_images=1):
G = Generator.from_pretrained("cropinky/projected_gan_impressionism")
return f"generating {number_of_images} images from {model}"
if __name__ == "__main__":
inputs = gr.inputs.Radio(["Abstract Expressionism", "Impressionism", "Cubism", "Minimalism", "Pop Art", "Color Field", "Hana Hanak houses"])
#outputs = gr.outputs.Image(label="Output Image")
outputs = "text"
title = "Projected GAN for painting generation"
description = "Choose your artistic direction "
article = "<p style='text-align: center'><a href='https://github.com/autonomousvision/projected_gan'>Official projected GAN github repo + paper</a></p>"
gr.Interface(image_generation, inputs, outputs, title=title, article = article,
description=description, analytics_enabled=False).launch(debug=True)
app, local_url, share_url = iface.launch()
|