Update app.py

app.py

@@ -0,0 +1,44 @@
+import streamlit as st
+from PIL import Image
+
+# UI Layout
+st.title("Florence-2 Demo")
+
+# Sidebar for model selection
+model_option = st.sidebar.selectbox(
+    "Select Model",
+    ("microsoft/Florence-2-large-ft", "microsoft/Florence-2-large", "microsoft/Florence-2-base-ft", "microsoft/Florence-2-base")
+)
+
+# Task prompt selection
+task_prompt = st.text_input("Task Prompt", value="Caption the image:")
+
+# Text input (optional)
+text_input = st.text_input("Text Input (Optional)")
+
+# Image upload
+uploaded_image = st.file_uploader("Upload an Image", type=["jpg", "jpeg", "png"])
+
+# Process the image if one is uploaded
+if uploaded_image is not None:
+    image = Image.open(uploaded_image)
+    st.image(image, caption="Uploaded Image", use_column_width=True)
+
+    # Run the model
+    if st.button("Run"):
+        # Assuming `run_example` function and others are defined as per the previous code snippet.
+        result = run_example(task_prompt, image, text_input, model_option)
+
+        # Display results
+        st.subheader("Generated Output")
+        st.write(result["text"])  # Display the generated text
+
+        # Display bounding boxes if available
+        if "bboxes" in result:
+            fig = plot_bbox(image, result)
+            st.pyplot(fig)
+
+        # Display polygons if available
+        if "polygons" in result:
+            processed_image = draw_polygons(image.copy(), result)
+            st.image(processed_image, caption="Image with Polygons", use_column_width=True)