Updated UI

app.py

@@ -27,20 +27,17 @@ densepose_weights = os.path.join(current_path, 'data', 'weights.pkl')
 
 texture_processor = TextureProcessor(densepose_config, densepose_weights)
 
-inputs = [
-    gr.inputs.Image(label='Person Image', type='numpy'),
-    gr.inputs.Image(label='Model Image (with clothes)', type='numpy')
-]
-
-outputs = gr.outputs.Image(label='Result Image', type='numpy')
-
-title = 'JustClothify'
-description = 'Upload an image of a person and an image of a model with clothes, the system will generate an image of a person wearing these clothes.'
-
-gr.Interface(
-    fn=image_processing, 
-    inputs=inputs, 
-    outputs=outputs, 
-    theme='soft',
-    title=title,
-    description=description).launch()
+title = '<h1 style="text-align:center">JustClothify</h1>'
+
+with gr.Blocks(theme='soft', title='AnswerMate') as blocks:
+    gr.HTML(title)
+    gr.Markdown('Upload an image of a person and an image of a model with clothes, the system will generate an image of a person wearing these clothes.')
+    with gr.Row():
+        person_image = gr.inputs.Image(label='Person Image', type='numpy')
+        model_image = gr.inputs.Image(label='Model Image (with clothes)', type='numpy')
+    process_button = gr.Button('Process')
+    outputs = gr.outputs.Image(label='Result Image', type='numpy')
+
+    process_button.click(fn=image_processing, inputs=[person_image, model_image], outputs=outputs)
+
+blocks.launch()