Update app.py

app.py

@@ -83,7 +83,7 @@ def generate_image(prompt, guidance_scale, num_inference_steps, randomize_seed,
         width=512,
         height=512
     ).images[0]
-
+    print(type(image),'image')
     return image, seed
 
 # Function to clear all arrays and reset the UI
@@ -94,7 +94,7 @@ def clear_arrays():
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("# Text-to-Image Generator with Object Addition")
+    gr.Markdown("# HiCo_T2I 512px")
 
     # Put prompt and submit button in the same row
     with gr.Group():
@@ -116,10 +116,10 @@ with gr.Blocks() as demo:
     )
 
     with gr.Row():
-        object_class_input = gr.Textbox(label="Object Class", placeholder="Enter object class (e.g., Object_1)")
+        object_class_input = gr.Textbox(label="Sub-caption", placeholder="Enter Sub-caption (e.g., apple)")
         bbox_input = gr.Textbox(label="Bounding Box (x1,y1,x2,y2)", placeholder="Enter bounding box coordinates")
 
-    add_button = gr.Button("Add Object")
+    add_button = gr.Button("Add")
 
     # Advanced settings in a collapsible accordion
     with gr.Accordion("Advanced Settings", open=False):