Update app.py

app.py

@@ -15,7 +15,7 @@ from enum import Enum
 
 api_key = os.getenv("FAI_API_KEY")
 api = os.getenv("FAI_API")
-#api_anime = os.getenv("ANIME_API")
+api_anime = os.getenv("FAI_API_2") #api_anime = os.getenv("ANIME_API")
 rmbgkey = os.getenv("RMBGKEY")
 
 #fore, prompt, intensity, mode, refprompt, isrmbg
@@ -216,9 +216,9 @@ def resize_to_fit(max_size, original_size):
     
 def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_type, steps, control_strength): 
 
-    max_size = (1500,1500)
+    max_size = (768,768)
     if prompt in prompt_list:
-        max_size = (1500,1500)
+        max_size = (768,768)
     if isrmbg:
         try:
             rmbgfore = rmbg(fore)
@@ -233,12 +233,12 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_typ
     image_width = size[0]
     image_height = size[1]
     
-    if size[0]*size[1]<=(768*768):
+    if size[0]*size[1]<=(512*512):
         gr.Warning("ℹ️ The input image resolution is low, it might lead to some deformation!")
         
     if size[0]*size[1]>(max_size[0]*max_size[1]):
         gr.Warning("ℹ️ The input image size is too big, I will lower it!")
-        image_width, image_height = resize_to_fit((1500,1500), (image_width, image_height))
+        image_width, image_height = resize_to_fit((768,768), (image_width, image_height))
         fore.resize(max_size)
 
     
@@ -250,17 +250,17 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_typ
         "prompt" : prompt,
         "mode" : mode,
         "intensity" : float(intensity),
-        "width" : 1500,
-        "height" : 1500,
+        "width" : 1000,
+        "height" : 1000,
         "refprompt" : refprompt,
         "first_stage_steps" : int(steps),
         "first_stage_strength" : float(control_strength),
-        "second_stage_steps" : 30
+        "second_stage_steps" : 20
     }
 
     # Select the endpoint based on the model type
     if model_type == "anime":
-        api_endpoint = api #to change to api_anime
+        api_endpoint = api_anime #to change to api_anime
     elif model_type == "realistic":
         api_endpoint = api
     else: