Drazcat-AI
/

florence_rp

Model card Files Files and versions

Drazcat-AI commited on Jan 16, 2025

Commit

fd12a8d

·

verified ·

1 Parent(s): 1a97d09

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -26,9 +26,9 @@ class EndpointHandler():
             num_beams=3,
             do_sample=False
         )
-        generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
-        parsed_answer = processor.post_process_generation(generated_text, task="<OD>", image_size=(image.width, image.height))
         return parsed_answer

             num_beams=3,
             do_sample=False
         )
+        generated_text = self.processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
+        parsed_answer = self.processor.post_process_generation(generated_text, task=prompt, image_size=(image.width, image.height))
         return parsed_answer