Spaces:
Sleeping
Sleeping
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
|