Update app.py

app.py

@@ -6,7 +6,7 @@ import scipy
 from PIL import Image
 import torch.nn as nn
 from transformers import Blip2Processor, Blip2ForConditionalGeneration, InstructBlipProcessor, InstructBlipForConditionalGeneration
-from my_model.object_detection import ObjectDetector
+from my_model.object_detection import detect_and_draw_objects
 
 def load_caption_model(blip2=False, instructblip=True):
 
@@ -96,15 +96,12 @@ def perform_object_detection(image, model_name, threshold):
     PIL.Image, str: The image with drawn bounding boxes and a string of detected objects.
     """
 
-    # Initialize the ObjectDetector
-    detector = ObjectDetector()
-    # Load the specified model
-    detector.load_model(model_name)
 
     try:
         # Perform object detection and draw bounding boxes
-        processed_image, detected_objects = detector.detect_and_draw_objects(image, model_name, threshold)
+        processed_image, detected_objects = detect_and_draw_objects(image, model_name, threshold)
         st.write("66666666666666")
+        st.write(detected_objects[:20])
         return processed_image, detected_objects
     except Exception as e:
         # Print the error for debugging