gen3

app.py

@@ -43,22 +43,13 @@ def infer(
     if prompt in [None, ""]:
         gr.Warning("⚠️ Please enter a prompt!")
         return None
-
-    if ref is not None:
-        editing_caption = "<image>" + prompt
-        input_image = ref
-        input_image = center_crop_arr_with_buckets(input_image, buckets=[512])
-    else:
-        editing_caption = prompt
-        input_image = None
-        img_cfg = 1.0
-
+    IMG_SIZE = 512
     image = pipeline.generate_image(
         captions=editing_caption,
         num_images_per_caption=1,
         positive_prompt=positive_prompt,
         negative_prompt=negative_prompt,
-        hw=(input_image.size[1], input_image.size[0]),
+        hw=(IMG_SIZE, IMG_SIZE),
         cfg=text_cfg,
         cfg_img=img_cfg,
         cfg_schedule="constant",