abmSS commited on
Commit
cc1e960
·
verified ·
1 Parent(s): 233793c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -1
app.py CHANGED
@@ -1,3 +1,17 @@
1
  import gradio as gr
2
 
3
- gr.load("models/enhanceaiteam/Flux-Uncensored-V2").launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
 
3
+ # Load the base model using the Hugging Face Inference API
4
+ def model_interface():
5
+ return gr.load("black-forest-labs/FLUX.1-dev", src="models")
6
+
7
+ # Load the LoRA
8
+ def lora_interface():
9
+ return gr.load("models/enhanceaiteam/Flux-Uncensored-V2", lora=True)
10
+
11
+ # Initialize the interfaces
12
+ base_model = model_interface()
13
+ lora_model = lora_interface()
14
+
15
+ # Launch the interfaces
16
+ base_model.launch()
17
+ lora_model.launch()