Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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("
|
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.
|
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:
|