comdoleger commited on
Commit
befaab6
1 Parent(s): 7c80faa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -253,9 +253,9 @@ def resize_to_fit(max_size, original_size):
253
 
254
  def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_type, steps, control_strength):
255
 
256
- max_size = (768,768)
257
  if prompt in prompt_list:
258
- max_size = (768,768)
259
  if isrmbg:
260
  try:
261
  rmbgfore = rmbg(fore)
@@ -270,6 +270,7 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_typ
270
  image_width = size[0]
271
  image_height = size[1]
272
 
 
273
  if size[0]*size[1]<=(512*512):
274
  gr.Warning("ℹ️ The input image resolution is low, it might lead to some deformation!")
275
 
 
253
 
254
  def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_type, steps, control_strength):
255
 
256
+ max_size = (1000,1000)
257
  if prompt in prompt_list:
258
+ max_size = (1200,1200)
259
  if isrmbg:
260
  try:
261
  rmbgfore = rmbg(fore)
 
270
  image_width = size[0]
271
  image_height = size[1]
272
 
273
+ gr.Warning(f"ℹ️ The input image size is: {size} ")
274
  if size[0]*size[1]<=(512*512):
275
  gr.Warning("ℹ️ The input image resolution is low, it might lead to some deformation!")
276