Update app.py

app.py

@@ -9,6 +9,19 @@ API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
 API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 
+def select_model(model_name):
+ models_list = [
+ "openskyml/dalle-3-xl",
+ "Linaqruf/animagine-xl-2.0",
+ "Lykon/dreamshaper-7",
+ "Linaqruf/animagine-xl",
+ "runwayml/stable-diffusion-v1-5",
+ "stabilityai/stable-diffusion-xl-base-1.0",
+ "prompthero/openjourney-v4",
+ "nerijs/pixel-art-xl",
+ "Linaqruf/anything-v3.0"
+ ]
+
 def query(prompt, is_negative=False, steps=1, cfg_scale=6, seed=None):
  payload = {
  "inputs": prompt,