Update app.py

app.py

@@ -266,7 +266,17 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_typ
  except:
  pass
  
- #fore = create_square_image(fore)
+ #
+ # Select the endpoint based on the model type
+ if model_type == "anime":
+ api_endpoint = api_anime #to change to api_anime
+ 
+ elif model_type == "realistic":
+ fore = create_square_image(fore)
+ api_endpoint = api_old
+ 
+ else:
+ raise ValueError("Invalid model type selected.")
  size = fore.size
  image_width = size[0]
  image_height = size[1]
@@ -297,25 +307,7 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_typ
  "second_stage_steps" : 20
  }
 
- # Select the endpoint based on the model type
- if model_type == "anime":
- api_endpoint = api_anime #to change to api_anime
- image = process(data, api_endpoint, api_key)
- elif model_type == "realistic":
- api_endpoint = api
- if data['mode']=="full":
- data['mode']="first-stage"
- image = process(data, api_endpoint, api_key)
- forestr = image_to_base64(image.convert("RGBA"))
- data['foreground_image64']=forestr
- data['mode']="refiner"
- api_endpoint = api_old
- image = process(data, api_endpoint, api_key)
- elif data['mode']=="refiner":
- api_endpoint = api_old
- image = process(data, api_endpoint, api_key)
- else:
- raise ValueError("Invalid model type selected.")
+
  #print(f"DATA: {data}")
  
  '''
@@ -329,7 +321,7 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_typ
  "refprompt" : " transparent glass " 
  }
  ''' 
- 
+ image = process(data, api_endpoint, api_key)
 
  return image