Update app.py

app.py

@@ -170,8 +170,8 @@ def resize_image(image, out_size):
 
 model_name = "vqgan_imagenet_f16_16384" 
 images_interval =  50
-width =  256
-height = 256
+width =  350
+height = 350
 init_image = ""
 seed = 42
 args = argparse.Namespace(
@@ -183,7 +183,7 @@ args = argparse.Namespace(
     clip_model='ViT-B/32',
     vqgan_config=f'{model_name}.yaml',
     vqgan_checkpoint=f'{model_name}.ckpt',
-    step_size=0.15,
+    step_size=0.12,
     cutn=4,
     cut_pow=1.,
     display_freq=images_interval,
@@ -197,7 +197,7 @@ perceptor = clip.load(args.clip_model, jit=False)[0].eval().requires_grad_(False
 def inference(text):
     texts = text
     target_images = ""
-    max_iterations = 150
+    max_iterations = 100
     model_names={"vqgan_imagenet_f16_16384": 'ImageNet 16384',"vqgan_imagenet_f16_1024":"ImageNet 1024", 'vqgan_openimages_f16_8192':'OpenImages 8912',
                     "wikiart_1024":"WikiArt 1024", "wikiart_16384":"WikiArt 16384", "coco":"COCO-Stuff", "faceshq":"FacesHQ", "sflckr":"S-FLCKR"}
     name_model = model_names[model_name]
@@ -346,7 +346,7 @@ def load_image( infilename ) :
 def throttled_inference(text):
     global inferences_running
     current = inferences_running
-    if current >= 2:
+    if current >= 1:
         print(f"Rejected inference when we already had {current} running")
         return load_image("./gpu.jpg")