Update app.py

app.py

@@ -48,11 +48,7 @@ inputs.append(gr.Number(
 names = ['prompt', 'negative_prompt', 'subject', 'number_of_outputs', 'number_of_images_per_pose', 'randomise_poses', 'output_format', 'output_quality', 'seed']
 
 outputs = []
-outputs.append(gr.Image())
-outputs.append(gr.Image())
-outputs.append(gr.Image())
-outputs.append(gr.Image())
-outputs.append(gr.Image())
+outputs.append(gr.Gallery())
 
 expected_outputs = len(outputs)
 def predict(request: gr.Request, *args, progress=gr.Progress(track_tqdm=True)):
@@ -88,13 +84,7 @@ def predict(request: gr.Request, *args, progress=gr.Progress(track_tqdm=True)):
         predict_outputs = parse_outputs(json_response["output"])
         processed_outputs = process_outputs(predict_outputs)
         difference_outputs = expected_outputs - len(processed_outputs)
-        # If less outputs than expected, hide the extra ones
-        if difference_outputs > 0:
-            extra_outputs = [gr.update(visible=False)] * difference_outputs
-            processed_outputs.extend(extra_outputs)
-        # If more outputs than expected, cap the outputs to the expected number
-        elif difference_outputs < 0:
-            processed_outputs = processed_outputs[:difference_outputs]
+
         
         return tuple(processed_outputs) if len(processed_outputs) > 1 else processed_outputs[0]
     else:
@@ -113,5 +103,5 @@ app = gr.Interface(
     description=model_description,
     allow_flagging="never",
 )
-app.launch(share=True)
+app.launch(share=False)