Update app.py

app.py

@@ -48,7 +48,7 @@ model = genai.GenerativeModel(model_name="gemini-1.5-flash-latest",
                               generation_config=generation_config,
                               safety_settings=safety_settings)
 
-input_prompt_template = """give me the info of the car (if an info is not available juste write "introuvable"):
+input_prompt_template = """give me the info of the car/truck (if an info is not available juste write "introuvable"):
 - plate:
 - model:
 - color: """
@@ -92,7 +92,7 @@ def detect_objects(image):
 
     # Loop through detections and filter only "car" class (ID 3 for COCO dataset)
     for score, label, box in zip(results["scores"], results["labels"], results["boxes"]):
-        if model.config.id2label[label.item()] == 'car' and score.item() > 0.9:
+        if (model.config.id2label[label.item()] == 'car' or model.config.id2label[label.item()] == 'truck' ) and score.item() > 0.9:
             box = [round(i, 2) for i in box.tolist()]
             # Crop the detected car
             cropped_car = image.crop(box)
@@ -102,7 +102,7 @@ def detect_objects(image):
 
             # Draw bounding box around the car
             draw.rectangle(box, outline="red", width=3)
-            draw.text((box[0], box[1]), f"Car: {round(score.item(), 2)}", fill="red")
+            draw.text((box[0], box[1]), f"véhicule: {round(score.item(), 2)}", fill="red")
     
     return image, detected_cars
 
@@ -135,7 +135,7 @@ def process_generate(files):
     car_info_list = []
     for car_bytes, box in detected_cars:
         car_info = generate_gemini_response(input_prompt_template, car_bytes)
-        car_info_list.append(f"Car at {box}:\n{car_info}\n")
+        car_info_list.append(f"véhicule aux coordonnées {box}:\n{car_info}\n")
 
     return detected_image, "\n".join(car_info_list)