Jonny001 commited on
Commit
83b2200
·
verified ·
1 Parent(s): f5bddfa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -6,11 +6,20 @@ model2 = gr.load("models/DiegoJR1973/NSFW-TrioHMH-Flux")
6
  model3 = gr.load("models/prashanth970/flux-lora-uncensored")
7
 
8
  def generate_images(text):
 
9
  result_image1 = model1(text)
10
  result_image2 = model2(text)
11
  result_image3 = model3(text)
12
 
13
 
 
 
 
 
 
 
 
 
14
  return result_image1, result_image2, result_image3
15
 
16
 
@@ -28,4 +37,4 @@ interface = gr.Interface(
28
  )
29
 
30
 
31
- interface.launch()
 
6
  model3 = gr.load("models/prashanth970/flux-lora-uncensored")
7
 
8
  def generate_images(text):
9
+
10
  result_image1 = model1(text)
11
  result_image2 = model2(text)
12
  result_image3 = model3(text)
13
 
14
 
15
+ if isinstance(result_image1, tuple):
16
+ result_image1 = result_image1[0]
17
+ if isinstance(result_image2, tuple):
18
+ result_image2 = result_image2[0]
19
+ if isinstance(result_image3, tuple):
20
+ result_image3 = result_image3[0]
21
+
22
+
23
  return result_image1, result_image2, result_image3
24
 
25
 
 
37
  )
38
 
39
 
40
+ interface.launch()