Update app.py

app.py

@@ -20,11 +20,6 @@ def answer_question(caption, detected_objects_str, question, model):
     answer = model.generate_answer(question, caption, detected_objects_str)
     return answer
 
-def get_caption(image):
-    return "Generated caption for the image"
-
-def free_gpu_resources():
-    pass
 
 # Sample images (assuming these are paths to your sample images)
 sample_images = ["Files/sample1.jpg", "Files/sample2.jpg", "Files/sample3.jpg", 
@@ -34,9 +29,12 @@ sample_images = ["Files/sample1.jpg", "Files/sample2.jpg", "Files/sample3.jpg",
 
 
 def analyze_image(image, model):
-    st.write("Analyzing . . .")
+    st.text("Cool image, let me analyze it..")
     caption = model.get_caption(image)
     image_with_boxes, detected_objects_str = model.detect_objects(image)
+    st.text("I am ready, let's talk!")
+    free_gpu_resources()
+    
     return caption, detected_objects_str
     
 
@@ -110,7 +108,7 @@ def run_inference():
     detection_model = st.selectbox(
         "Choose a model for object detection:",
         ["yolov5", "detic"],
-        index=0
+        index=1  # "detic" is selected by default
     )
 
     default_confidence = 0.2 if detection_model == "yolov5" else 0.4
@@ -135,14 +133,16 @@ def run_inference():
             st.session_state['kbvqa'] = None
 
         if st.button(button_label):
+            free_gpu_resources()
             if st.session_state['kbvqa'] is not None and not settings_changed:
                 st.write("Model already loaded.")
             else:
-                st.text("Loading the model will take no more than a few minutes . .")
+                st.text("Loading the model will take no more than a few minutes dpending on the hardware configuration . .")
                 st.session_state['kbvqa'] = prepare_kbvqa_model(detection_model)
                 st.session_state['kbvqa'].detection_confidence = confidence_level
                 st.session_state['model_settings'] = {'detection_model': detection_model, 'confidence_level': confidence_level}
                 st.write("Model is ready for inference.")
+                free_gpu_resources()
 
         if st.session_state['kbvqa']:
             image_qa_app(st.session_state['kbvqa'])