Update app.py

app.py

@@ -42,16 +42,17 @@ def process_image_and_text(image, text):
     # Only get generated tokens; decode them to text
     generated_tokens = output[0, inputs['input_ids'].size(1):]
     generated_text = processor.tokenizer.decode(generated_tokens, skip_special_tokens=True)
-    pretty_text = pprint.pp(generated_text)
 
-    return pretty_text
+    return generated_text
 
 def chatbot(image, text, history):
     if image is None:
         return history + [("Please upload an image first.", None)]
 
     response = process_image_and_text(image, text)
-    history.append((text, response))
+    pretty_response = pprint.pp(response)
+
+    history.append((text, pretty_response))
     return history
 
 # Define the Gradio interface