Update app.py

line 19 + seed
line 102 + api_name
line 103 +seed

app.py

@@ -16,7 +16,7 @@ def infer(prompt, seed, randomize_seed, width, height, guidance_scale, num_infer
 		api_name="/infer"
     )
 
-    return result
+    return result seed
 
     
 css="""
@@ -31,7 +31,7 @@ with gr.Blocks(css=css) as demo:
     
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""
-        #DiffusionLab Beta
+        FallnAI: DiffusionLab Beta 
         """)
         
         with gr.Row():
@@ -99,8 +99,8 @@ with gr.Blocks(css=css) as demo:
 
     run_button.click(
         fn = infer,
-        inputs = [prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
-        outputs = [result]
+        inputs = [prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps,api_name],
+        outputs = [result, seed]
     )
 
 demo.queue().launch()