Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ import gradio as gr
|
|
| 4 |
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/
|
| 8 |
model5 = gr.load("models/prashanth970/flux-lora-uncensored")
|
| 9 |
|
| 10 |
def generate_images(text, selected_model):
|
|
@@ -14,7 +14,7 @@ def generate_images(text, selected_model):
|
|
| 14 |
model = model2
|
| 15 |
elif selected_model == "Model 3 (WBMH-v1.1)":
|
| 16 |
model = model3
|
| 17 |
-
elif selected_model == "Model 4 (
|
| 18 |
model = model4
|
| 19 |
elif selected_model == "Model 5 (Lora Uncensored)":
|
| 20 |
model = model5
|
|
@@ -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 (
|
| 38 |
label="Select Model (Try All Models & Get Different Results)",
|
| 39 |
value="Model 1 (NSFW Master)",
|
| 40 |
),
|
|
|
|
| 4 |
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/prashanth970/flux-lora-uncensored")
|
| 9 |
|
| 10 |
def generate_images(text, selected_model):
|
|
|
|
| 14 |
model = model2
|
| 15 |
elif selected_model == "Model 3 (WBMH-v1.1)":
|
| 16 |
model = model3
|
| 17 |
+
elif selected_model == "Model 4 (EXD)":
|
| 18 |
model = model4
|
| 19 |
elif selected_model == "Model 5 (Lora Uncensored)":
|
| 20 |
model = model5
|
|
|
|
| 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 (Lora Uncensored)"],
|
| 38 |
label="Select Model (Try All Models & Get Different Results)",
|
| 39 |
value="Model 1 (NSFW Master)",
|
| 40 |
),
|