kh-CHEUNG commited on
Commit
c799276
·
verified ·
1 Parent(s): c032942

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -41,7 +41,7 @@ def unnormalize_box(box, image_width, image_height):
41
  processor = AutoProcessor.from_pretrained("microsoft/udop-large", apply_ocr=True)
42
  model = UdopForConditionalGeneration.from_pretrained("microsoft/udop-large")
43
 
44
- st.title("CIC Demo (by ITT)")
45
  st.write("Upload and Select a document (/an image) to test the model.")
46
 
47
  #2 column layout
@@ -67,7 +67,7 @@ with col2:
67
  # question = "Question answering. How many unsafe practice of Lifting Operation?"
68
  default_question = "Is this a Lifting Operation scene?"
69
  task_type = st.selectbox("Question Type:", ("Classification", "Question Answering", "Layout Analysis"))
70
- question_text = st.text_input("Prompt:", placeholder=default_question)
71
  if question_text is not None:
72
  question = task_type + ". " + question_text
73
  else:
 
41
  processor = AutoProcessor.from_pretrained("microsoft/udop-large", apply_ocr=True)
42
  model = UdopForConditionalGeneration.from_pretrained("microsoft/udop-large")
43
 
44
+ st.title("GenAI Demo (by ITT)")
45
  st.write("Upload and Select a document (/an image) to test the model.")
46
 
47
  #2 column layout
 
67
  # question = "Question answering. How many unsafe practice of Lifting Operation?"
68
  default_question = "Is this a Lifting Operation scene?"
69
  task_type = st.selectbox("Question Type:", ("Classification", "Question Answering", "Layout Analysis"))
70
+ question_text = st.text_area("Prompt:", placeholder=default_question)
71
  if question_text is not None:
72
  question = task_type + ". " + question_text
73
  else: