Update app.py

app.py

@@ -6,7 +6,9 @@ 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") 
+imgpipe = pipeline("zero-shot-image-classification", model="google/siglip-so400m-patch14-384")
+
+
 st.title("NLP APP")
 option = st.sidebar.selectbox(
     "Choose a task",
@@ -33,10 +35,14 @@ elif option == "Object Detection":
     st.title("Welcome to object detection")
 
     uploaded_files = st.file_uploader("Choose a image file",type=["jpg","jpeg"])
-
+    text=st.text_area("Enter possible class names(comma separated")
+    candidate_lables=[t.strip() for t in text.split(',')]
     if uploaded_files is not None:
         Image=Image.open(uploaded_files)
-        st.write(objpipe(Image)[0]["label"])
+        
+        outputs = imgpipe(uploaded_files,candidate_lables)
+
+        st.write(output["label"])
         
 
 else: