Update app.py

app.py

@@ -20,7 +20,6 @@ model = ImageCaptioningModel(training_cfg=training_cfg)
 model.to(device)
 model.eval()
 
-# Load checkpoint from the repo root
 CHECKPOINT_PATH = "best_model.pt"
 state_dict = torch.load(CHECKPOINT_PATH, map_location=device)
 model.load_state_dict(state_dict)
@@ -54,7 +53,7 @@ async def caption_image(file: UploadFile = File(...)) -> JSONResponse:
         captions: List[str] = model.generate(
             images=tensor,
             max_length=50,
-            num_beams=1,  # deterministic greedy decoding
+            num_beams=1,
         )
 
     return JSONResponse({"caption": captions[0]})