Update app.py

app.py

@@ -35,7 +35,7 @@ def infer(
           do_sample=False
       )
     result = processor.batch_decode(generated_ids, skip_special_tokens=True)
-    return result[0][len(text):]
+    return result[0][len(text):].lstrip("\n")
 
 ##### Parse segmentation output tokens into masks
 ##### Also returns bounding boxes with their labels
@@ -85,7 +85,7 @@ for detailed information how to use and fine-tune PaliGemma models.
 
 with gr.Blocks(css="style.css") as demo:
   gr.Markdown(INTRO_TEXT)
-  with gr.Tab("Conversation"):
+  with gr.Tab("Text Generation"):
     with gr.Column():
         image = gr.Image(type="pil")
         text_input = gr.Text(label="Input Text")