salso commited on
Commit
404d1fc
1 Parent(s): ecbb850

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,7 +15,7 @@ from enum import Enum
15
 
16
  api_key = os.getenv("FAI_API_KEY")
17
  api = os.getenv("FAI_API")
18
- api_anime = os.getenv("ANIME_API")
19
  rmbgkey = os.getenv("RMBGKEY")
20
 
21
  #fore, prompt, intensity, mode, refprompt, isrmbg
@@ -259,7 +259,7 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_typ
259
 
260
  # Select the endpoint based on the model type
261
  if model_type == "anime":
262
- api_endpoint = api_anime
263
  elif model_type == "realistic":
264
  api_endpoint = api
265
  else:
 
15
 
16
  api_key = os.getenv("FAI_API_KEY")
17
  api = os.getenv("FAI_API")
18
+ #api_anime = os.getenv("ANIME_API")
19
  rmbgkey = os.getenv("RMBGKEY")
20
 
21
  #fore, prompt, intensity, mode, refprompt, isrmbg
 
259
 
260
  # Select the endpoint based on the model type
261
  if model_type == "anime":
262
+ api_endpoint = api #to change
263
  elif model_type == "realistic":
264
  api_endpoint = api
265
  else: