@@ -70,6 +70,7 @@ class Nanobanana(SimpleService): except ModelError as exc: if exc.prediction.error == 'No image content found in response': raise ImageContentNotFound + raise process_time = timedelta(seconds=(time.time() - start_time)) self.handle_invoice(input_message.content_object.model, version=version, resolution=resolution) msgs = self.save_results(input_message.content, image, process_time, save)