Spaces:
Sleeping
Sleeping
Mathdesenvnonimate
commited on
Commit
•
99b54b6
1
Parent(s):
7e0b7b0
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ from PIL import Image
|
|
8 |
# List of available models
|
9 |
|
10 |
list_models = [
|
11 |
-
"FunkoPOP", "
|
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 == "
|
20 |
-
API_URL = "https://api-inference.huggingface.co/models/
|
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 |
|