after black lint

Luodian · Sep 11, 2023 · ac0cfcd · ac0cfcd
1 parent dff80c3
commit ac0cfcd
Showing 1 changed file with 1 addition and 11 deletions.
diff --git a/pipeline/evaluation/models/otter.py b/pipeline/evaluation/models/otter.py
@@ -51,17 +51,7 @@ def get_response(image: Image.Image, prompt: str, model=None, image_processor=No
         num_beams=3,
         no_repeat_ngram_size=3,
     )
-    parsed_output = (
-        model.text_tokenizer.decode(generated_text[0])
-        .split("<answer>")[-1]
-        .lstrip()
-        .rstrip()
-        .split("<|endofchunk|>")[0]
-        .lstrip()
-        .rstrip()
-        .lstrip('"')
-        .rstrip('"')
-    )
+    parsed_output = model.text_tokenizer.decode(generated_text[0]).split("<answer>")[-1].lstrip().rstrip().split("<|endofchunk|>")[0].lstrip().rstrip().lstrip('"').rstrip('"')
     return parsed_output