Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -1,18 +1,23 @@
|
|
1 |
-
#model1 = gr.load("models/pimpilikipilapi1/NSFW_master")
|
2 |
-
#model2 = gr.load("models/DiegoJR1973/NSFW-TrioHMH-Flux")
|
3 |
-
#model3 = gr.load("models/prashanth970/flux-lora-uncensored")
|
4 |
-
|
5 |
import gradio as gr
|
6 |
|
7 |
|
8 |
-
model1 = gr.load("models/
|
9 |
-
model2 = gr.load("models/
|
|
|
|
|
|
|
10 |
|
11 |
def generate_images(text, selected_model):
|
12 |
if selected_model == "Model 1 (NSFW Master)":
|
13 |
model = model1
|
14 |
-
elif selected_model == "Model 2 (Flux
|
15 |
model = model2
|
|
|
|
|
|
|
|
|
|
|
|
|
16 |
else:
|
17 |
return "Invalid model selection."
|
18 |
|
@@ -29,9 +34,9 @@ interface = gr.Interface(
|
|
29 |
inputs=[
|
30 |
gr.Textbox(label="Type here your imagination:", placeholder="Type your prompt..."),
|
31 |
gr.Radio(
|
32 |
-
["Model 1 (NSFW Master)", "Model 2 (Flux Lora Uncensored)"],
|
33 |
-
label="Select Model",
|
34 |
-
value="Model
|
35 |
),
|
36 |
],
|
37 |
outputs=[
|
@@ -41,6 +46,7 @@ interface = gr.Interface(
|
|
41 |
],
|
42 |
theme="Yntec/HaleyCH_Theme_Orange",
|
43 |
description="⚠ Sorry for the inconvenience. The models are currently running on the CPU, which might affect performance. We appreciate your understanding.",
|
|
|
44 |
)
|
45 |
|
46 |
interface.launch()
|
|
|
|
|
|
|
|
|
|
|
1 |
import gradio as gr
|
2 |
|
3 |
|
4 |
+
model1 = gr.load("models/Jonny001/NSFW_master")
|
5 |
+
model2 = gr.load("models/xey/sldr_flux_nsfw_v2-studio")
|
6 |
+
model3 = gr.load("models/lexa862/NSFWmodel")
|
7 |
+
model4 = gr.load("models/Keltezaa/flux_pussy_NSFW")
|
8 |
+
model5 = gr.load("models/prashanth970/flux-lora-uncensored")
|
9 |
|
10 |
def generate_images(text, selected_model):
|
11 |
if selected_model == "Model 1 (NSFW Master)":
|
12 |
model = model1
|
13 |
+
elif selected_model == "Model 2 (Sldr Flux NSFW)":
|
14 |
model = model2
|
15 |
+
elif selected_model == "Model 3 (Lexa NSFW)":
|
16 |
+
model = model3
|
17 |
+
elif selected_model == "Model 4 (Flux NSFW)":
|
18 |
+
model = model4
|
19 |
+
elif selected_model == "Model 4 (Lora Uncensored)":
|
20 |
+
model = model5
|
21 |
else:
|
22 |
return "Invalid model selection."
|
23 |
|
|
|
34 |
inputs=[
|
35 |
gr.Textbox(label="Type here your imagination:", placeholder="Type your prompt..."),
|
36 |
gr.Radio(
|
37 |
+
["Model 1 (NSFW Master)", "Model 2 (Sldr Flux NSFW)", "Model 3 (Lexa NSFW)", "Model 4 (Flux NSFW)", "Model 5 (Lora Uncensored)"],
|
38 |
+
label="Select Model (Try All Models & Get Different Results)",
|
39 |
+
value="Model 1 (NSFW Master)",
|
40 |
),
|
41 |
],
|
42 |
outputs=[
|
|
|
46 |
],
|
47 |
theme="Yntec/HaleyCH_Theme_Orange",
|
48 |
description="⚠ Sorry for the inconvenience. The models are currently running on the CPU, which might affect performance. We appreciate your understanding.",
|
49 |
+
cache_examples=False,
|
50 |
)
|
51 |
|
52 |
interface.launch()
|