Update app.py

app.py

@@ -6,7 +6,7 @@ from PIL import Image
 
 pipe = pipeline("summarization", model="google/pegasus-xsum")
 agepipe = pipeline("image-classification", model="dima806/facial_age_image_detection")
-
+objpipe = pipeline("zero-shot-object-detection", model="google/owlvit-base-patch32") 
 st.title("NLP APP")
 option = st.sidebar.selectbox(
     "Choose a task",
@@ -27,7 +27,17 @@ elif option == "Age Detection":
 
     if uploaded_files is not None:
         Image=Image.open(uploaded_files)
+
         st.write(agepipe(Image)[0]["label"])
+elif option == "Object Detection":
+    st.title("Welcome to object detection")
+
+    uploaded_files = st.file_uploader("Choose a image file",type="jpg")
+
+    if uploaded_files is not None:
+        Image=Image.open(uploaded_files)
+        st.write(objpipe(Image)[0]["label"])
+        
 
 else:
     st.title("None")