app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import time
 
-models =[
+models = [
     "enhanceaiteam/Flux-uncensored",
     "adirik/flux-cinestill",
     "kudzueye/boreal-flux-dev-v2",
@@ -16,13 +16,12 @@ model_functions = {}
 model_idx = 1
 for model_path in models:
     try:
-        model_functions[model_idx] = gr.Interface.load(f"models/{model_path}", live=False, preprocess=True, postprocess=False)
+        model_functions[model_idx] = gr.load(f"models/{model_path}", live=False)
     except Exception as error:
         def the_fn(txt):
             return None
         model_functions[model_idx] = gr.Interface(fn=the_fn, inputs=["text"], outputs=["image"])
-    model_idx+=1
-
+    model_idx += 1
 
 def send_it_idx(idx):
     def send_it_fn(prompt):
@@ -38,10 +37,9 @@ def clear_it(val):
         val = 0
     else:
         val = 0
-        pass
     return val
 
-def all_task_end(cnt,t_stamp):
+def all_task_end(cnt, t_stamp):
     to = t_stamp + 360
     et = time.time()
     if et > to and t_stamp != 0:
@@ -53,11 +51,9 @@ def all_task_end(cnt,t_stamp):
         else:
             d = gr.update(value=0)
         tog = gr.update(value=0)
-        pass
     return d, tog
 
 def all_task_start():
-    print("\n\n\n\n\n\n\n")
     t = time.gmtime()
     t_stamp = time.time()
     current_time = time.strftime("%H:%M:%S", t)
@@ -67,17 +63,14 @@ def clear_fn():
     nn = len(models)
     return tuple([None, *[None for _ in range(nn)]])
 
-
-
 with gr.Blocks(title="SD Models") as my_interface:
     with gr.Column(scale=12):
         with gr.Row():
-            with gr.Row(scale=6):
-                primary_prompt=gr.Textbox(label="Prompt", value="")
-            with gr.Row(scale=6):
-                with gr.Row():
-                    run=gr.Button("Run",variant="primary")
-                    clear_btn=gr.Button("Clear")
+            with gr.Row():
+                primary_prompt = gr.Textbox(label="Prompt", value="")
+            with gr.Row():
+                run = gr.Button("Run", variant="primary")
+                clear_btn = gr.Button("Clear")
         with gr.Row():
             sd_outputs = {}
             model_idx = 1
@@ -85,22 +78,19 @@ with gr.Blocks(title="SD Models") as my_interface:
                 with gr.Column(scale=3, min_width=200):
                     with gr.Box():
                         sd_outputs[model_idx] = gr.Image(label=model_path)
-                    pass
                 model_idx += 1
-                pass
-            pass
-
         with gr.Row(visible=False):
-            start_box=gr.Number(interactive=False)
-            end_box=gr.Number(interactive=False)
-            tog_box=gr.Textbox(value=0,interactive=False)
+            start_box = gr.Number(interactive=False)
+            end_box = gr.Number(interactive=False)
+            tog_box = gr.Textbox(value=0, interactive=False)
 
         start_box.change(
             all_task_end,
             [start_box, end_box],
             [start_box, tog_box],
             every=1,
-            show_progress=True)
+            show_progress=True
+        )
 
         primary_prompt.submit(all_task_start, None, [start_box, end_box, tog_box])
         run.click(all_task_start, None, [start_box, end_box, tog_box])
@@ -109,20 +99,19 @@ with gr.Blocks(title="SD Models") as my_interface:
         for model_path in models:
             runs_dict[model_idx] = run.click(model_functions[model_idx], inputs=[primary_prompt], outputs=[sd_outputs[model_idx]])
             model_idx += 1
-            pass
-        pass
 
         clear_btn.click(
             clear_fn,
             None,
             [primary_prompt, *list(sd_outputs.values())],
-            cancels=[*list(runs_dict.values())])
+            cancels=[*list(runs_dict.values())]
+        )
         tog_box.change(
             clear_it,
             tog_box,
             tog_box,
-            cancels=[*list(runs_dict.values())])
+            cancels=[*list(runs_dict.values())]
+        )
 
 my_interface.queue(concurrency_count=100, status_update_rate=1)
-my_interface.launch(inline=True, show_api=False)
- 
+my_interface.launch(inline=True, show_api=False)