Ahsen Khaliq commited on
Commit
1880628
β€’
1 Parent(s): d49d4ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -203,7 +203,9 @@ def inference(text, seed, step_size, max_iterations, width, height, init_image,
203
  size=[width, height]
204
  texts = text
205
  init_weight=init_weight
206
- init_image = init_image.name
 
 
207
  target_images = ""
208
  max_iterations = max_iterations
209
  model_names={"vqgan_imagenet_f16_16384": 'ImageNet 16384',"vqgan_imagenet_f16_1024":"ImageNet 1024", 'vqgan_openimages_f16_8192':'OpenImages 8912',
 
203
  size=[width, height]
204
  texts = text
205
  init_weight=init_weight
206
+ if init_image != None
207
+ init_image = init_image.name
208
+ init_image = ""
209
  target_images = ""
210
  max_iterations = max_iterations
211
  model_names={"vqgan_imagenet_f16_16384": 'ImageNet 16384',"vqgan_imagenet_f16_1024":"ImageNet 1024", 'vqgan_openimages_f16_8192':'OpenImages 8912',