Nougat

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

import app


def get_image_predict(img: Image):
    model_output = app.model.inference(img)

    predictions = []
    for j, output in enumerate(model_output["predictions"]):
        if model_output["repeats"][j] is not None:
            if model_output["repeats"][j] > 0:
                disclaimer = "\n\n+++ ==WARNING: Truncated because of repetitions==\n%s\n+++\n\n"
            else:
                disclaimer = (
                    "\n\n+++ ==ERROR: No output for this page==\n%s\n+++\n\n"
                )
            rest = app.close_envs(model_output["repetitions"][j]).strip()
            if len(rest) > 0:
                disclaimer = disclaimer % rest
            else:
                disclaimer = ""
        else:
            disclaimer = ""

        predictions.append(app.markdown_compatible(output) + disclaimer)


    return "".join(predictions).strip()