Update app.py

app.py

@@ -36,6 +36,10 @@ if uploaded_file is not None:
         messages, tokenize=False, add_generation_prompt=True
     )
 
+    # Debugging: Display the generated text
+    st.write("Generated text for processing:")
+    st.write(text)
+
     # Pass the image to the processor
     inputs = processor(
         text=[text],
@@ -45,6 +49,10 @@ if uploaded_file is not None:
     )
     inputs = inputs.to("cpu")
 
+    # Debugging: Display the inputs
+    st.write("Inputs for the model:")
+    st.write(inputs)
+
     # Inference: Generation of the output
     generated_ids = model.generate(**inputs, max_new_tokens=128)
     generated_ids_trimmed = [
@@ -54,5 +62,9 @@ if uploaded_file is not None:
         generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
     )
     
+    # Debugging: Display the raw output text
+    st.write("Raw output text:")
+    st.write(output_text)
+
     st.write("Description:")
-    st.write(output_text)
+    st.write(output_text[0])