Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ model1 = gr.load("models/Jonny001/NSFW_master")
|
|
5 |
model2 = gr.load("models/Jonny001/Alita-v1")
|
6 |
model3 = gr.load("models/Jonny001/WBMH-v1.1")
|
7 |
model4 = gr.load("models/Jonny001/EXD-v1")
|
8 |
-
model5 = gr.load("models/
|
9 |
|
10 |
def generate_images(text, selected_model):
|
11 |
if selected_model == "Model 1 (NSFW Master)":
|
@@ -16,7 +16,7 @@ def generate_images(text, selected_model):
|
|
16 |
model = model3
|
17 |
elif selected_model == "Model 4 (EXD)":
|
18 |
model = model4
|
19 |
-
elif selected_model == "Model 5 (
|
20 |
model = model5
|
21 |
else:
|
22 |
return "Invalid model selection."
|
@@ -34,7 +34,7 @@ interface = gr.Interface(
|
|
34 |
inputs=[
|
35 |
gr.Textbox(label="Type here your imagination:", placeholder="Type your prompt..."),
|
36 |
gr.Radio(
|
37 |
-
["Model 1 (NSFW Master)", "Model 2 (Alita)", "Model 3 (WBMH-v1.1)", "Model 4 (EXD)", "Model 5 (
|
38 |
label="Select Model (Try All Models & Get Different Results)",
|
39 |
value="Model 1 (NSFW Master)",
|
40 |
),
|
|
|
5 |
model2 = gr.load("models/Jonny001/Alita-v1")
|
6 |
model3 = gr.load("models/Jonny001/WBMH-v1.1")
|
7 |
model4 = gr.load("models/Jonny001/EXD-v1")
|
8 |
+
model5 = gr.load("models/Jonny001/R.I.O.W")
|
9 |
|
10 |
def generate_images(text, selected_model):
|
11 |
if selected_model == "Model 1 (NSFW Master)":
|
|
|
16 |
model = model3
|
17 |
elif selected_model == "Model 4 (EXD)":
|
18 |
model = model4
|
19 |
+
elif selected_model == "Model 5 (R.I.O.W)":
|
20 |
model = model5
|
21 |
else:
|
22 |
return "Invalid model selection."
|
|
|
34 |
inputs=[
|
35 |
gr.Textbox(label="Type here your imagination:", placeholder="Type your prompt..."),
|
36 |
gr.Radio(
|
37 |
+
["Model 1 (NSFW Master)", "Model 2 (Alita)", "Model 3 (WBMH-v1.1)", "Model 4 (EXD)", "Model 5 (R.I.O.W)"],
|
38 |
label="Select Model (Try All Models & Get Different Results)",
|
39 |
value="Model 1 (NSFW Master)",
|
40 |
),
|