Update app.py

app.py

@@ -6,7 +6,7 @@ import streamlit as st
 # processor = AutoProcessor.from_pretrained("microsoft/udop-large", apply_ocr=True)
 # model = UdopForConditionalGeneration.from_pretrained("microsoft/udop-large")
 
-st.title("Document (/Image) Selection and Upload")
+st.title("CIC Demo (by ITT)")
 st.write("Select or upload a document (/an image) to test the model.")
 
 # Image selection
@@ -18,9 +18,10 @@ if selected_file is not None and selected_file != "None":
     st.image(selected_file, caption="Selected Image")
 
 # Model testing button
-if st.button("Test Model") and selected_file != "None":
+testButton = st.button("Test Model")
+if testButton and selected_file != "None":
     st.write("Testing the model with the selected image...")
-elif st.button("Test Model") and selected_file == "None":
+elif testButton and selected_file == "None":
     st.write("Please upload and select an image.")
 
 # encoding = processor(image, question, words, boxes=boxes, return_tensors="pt")