Mathdesenvnonimate commited on
Commit
99b54b6
1 Parent(s): 7e0b7b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,7 +8,7 @@ from PIL import Image
8
  # List of available models
9
 
10
  list_models = [
11
- "FunkoPOP", "SD 1.5",
12
  ]
13
 
14
  # Function to generate images from text
@@ -16,8 +16,8 @@ def generate_txt2img(current_model, prompt, is_negative=False, image_style="None
16
 
17
  if current_model == "FunkoPOP":
18
  API_URL = "https://api-inference.huggingface.co/models/ProomptEngineer/pe-funko-pop-diffusion-style"
19
- elif current_model == "SD 1.5":
20
- API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
21
  API_TOKEN = os.environ.get("HF_READ_TOKEN")
22
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
23
 
 
8
  # List of available models
9
 
10
  list_models = [
11
+ "FunkoPOP", "Funko",
12
  ]
13
 
14
  # Function to generate images from text
 
16
 
17
  if current_model == "FunkoPOP":
18
  API_URL = "https://api-inference.huggingface.co/models/ProomptEngineer/pe-funko-pop-diffusion-style"
19
+ elif current_model == "Funko":
20
+ API_URL = "https://api-inference.huggingface.co/models/ProomptEngineer/pe-funko-pop-diffusion-style"
21
  API_TOKEN = os.environ.get("HF_READ_TOKEN")
22
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
23