Update app.py

app.py

@@ -42,15 +42,15 @@ processor = AutoProcessor.from_pretrained("microsoft/udop-large", apply_ocr=True
 model = UdopForConditionalGeneration.from_pretrained("microsoft/udop-large")
 
 st.title("GenAI Demo (by ITT)")
-st.write("Upload and Select a document (/an image) to test the model.")
+st.text("Upload and Select a document (/an image) to test the model.")
 
 #2 column layout
 col1, col2 = st.columns(2)
 
 with col1:  
     # File selection
-    uploaded_files = st.file_uploader("Upload document(s) [/image(s)]:", type=["docx", "pdf", "pptx", "jpg", "jpeg", "png"], accept_multiple_files=True)
-    selected_file = st.selectbox("Select a document (/an image):", uploaded_files, format_func=lambda file: file.name if file else "None")
+    uploaded_files = st.file_uploader("Upload document(s) [/image(s)]:", type=["docx", "pdf", "pptx", "jpg", "jpeg", "png"], accept_multiple_files=True, , key="fileUpload")
+    selected_file = st.selectbox("Select a document (/an image):", uploaded_files, format_func=lambda file: file.name if file else "None", key="fileSelect")
     
     # Display selected file
     if selected_file is not None and selected_file != "None":
@@ -66,15 +66,15 @@ with col2:
     ## Question (/Prompt)
     # question = "Question answering. How many unsafe practice of Lifting Operation?"
     default_question = "Is this a Lifting Operation scene?"
-    task_type = st.selectbox("Question Type:", ("Classification", "Question Answering", "Layout Analysis"))
-    question_text = st.text_area("Prompt:", placeholder=default_question)
+    task_type = st.selectbox("Question Type:", ("Classification", "Question Answering", "Layout Analysis"), index=1, key="taskSelect")
+    question_text = st.text_area("Prompt:", placeholder=default_question, key="questionInput")
     if question_text is not None: 
         question = task_type + ". " + question_text
     else:
         question = task_type + ". " + default_question 
     
     ## Test button
-    testButton = st.button("Test Model")
+    testButton = st.button("Test Model", key="testStart")
     
     ## Perform Model Testing when Image is uploaded and selected as well as Test button is pressed
     if testButton and selected_file != "None":