Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ os.environ["OMP_NUM_THREADS"] = str(os.cpu_count())
|
|
7 |
torch.set_num_threads(os.cpu_count())
|
8 |
|
9 |
model1 = gr.load("models/prithivMLmods/SD3.5-Turbo-Realism-2.0-LoRA")
|
10 |
-
model2 = gr.load("models/
|
11 |
|
12 |
stop_event = threading.Event()
|
13 |
|
@@ -16,7 +16,7 @@ def generate_images(text, selected_model):
|
|
16 |
|
17 |
if selected_model == "Model 1 (Turbo Realism)":
|
18 |
model = model1
|
19 |
-
elif selected_model == "Model 2 (
|
20 |
model = model2
|
21 |
else:
|
22 |
return ["Invalid model selection."] * 3
|
|
|
7 |
torch.set_num_threads(os.cpu_count())
|
8 |
|
9 |
model1 = gr.load("models/prithivMLmods/SD3.5-Turbo-Realism-2.0-LoRA")
|
10 |
+
model2 = gr.load("models/Purz/face-projection")
|
11 |
|
12 |
stop_event = threading.Event()
|
13 |
|
|
|
16 |
|
17 |
if selected_model == "Model 1 (Turbo Realism)":
|
18 |
model = model1
|
19 |
+
elif selected_model == "Model 2 (Face Projection)":
|
20 |
model = model2
|
21 |
else:
|
22 |
return ["Invalid model selection."] * 3
|