waloneai commited on
Commit
e358038
·
verified ·
1 Parent(s): 7339be3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/Purz/face-projection")
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 (Face Projection)":
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/black-forest-labs/FLUX.1-schnell")
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 (Schnell)":
20
  model = model2
21
  else:
22
  return ["Invalid model selection."] * 3