Update app.py

app.py

@@ -8,7 +8,7 @@ def generate_text(prompt, image):
     # Generate text description
     prompt = f"USER: <image>\n{prompt}\nASSISTANT:"
     text_description = pipe(images=image, prompt=prompt,generate_kwargs={"max_new_tokens": 200}) # Batch for efficiency
-    return text_description[0]['generated_text']
+    return text_description[0]['generated_text'].split("\nASSISTANT:")[-1]
 
 # Create Gradio interface
 inputs = [gr.Textbox(label="Input Text"), gr.Image(type="pil")]  # Input for uploading an image