Cropinky commited on
Commit
c861fe5
·
1 Parent(s): f1186da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -9,12 +9,11 @@ import PIL
9
  from image_generator import generate_images
10
 
11
  def image_generation(model, number_of_images=1):
12
- G = MyGenerator.from_pretrained("Cropinky/projected_gan_impressionism")
13
  img = generate_images(model)
14
  #return f"generating {number_of_images} images from {model}"
15
  return img
16
  if __name__ == "__main__":
17
- description = "TODO: when generating only 1 image use an esrgan to increase its resolution \n TODO: allow generation of multiple images"
18
  inputs = gr.inputs.Radio(["Abstract Expressionism", "Impressionism", "Cubism", "Pop Art", "Color Field", "Hana Hanak houses", "Hana Hanak houses - abstract expressionism", "Hana Hanak houses - color field"])
19
  outputs = gr.outputs.Image(label="Generated Image", type="pil")
20
  #outputs = "text"
 
9
  from image_generator import generate_images
10
 
11
  def image_generation(model, number_of_images=1):
 
12
  img = generate_images(model)
13
  #return f"generating {number_of_images} images from {model}"
14
  return img
15
  if __name__ == "__main__":
16
+ description = "TODO: when generating only 1 image use an esrgan to increase its resolution \n TODO: allow generation of multiple images TODO: walk through input space video i have exams now c u in 2 weeks (:"
17
  inputs = gr.inputs.Radio(["Abstract Expressionism", "Impressionism", "Cubism", "Pop Art", "Color Field", "Hana Hanak houses", "Hana Hanak houses - abstract expressionism", "Hana Hanak houses - color field"])
18
  outputs = gr.outputs.Image(label="Generated Image", type="pil")
19
  #outputs = "text"