ECOFRI
/

CXR-LLAVA-v2

Feature Extraction

Model card Files Files and versions Community

ECOFRI commited on Jan 18

Commit

341800b

•

1 Parent(s): c7ca54d

Upload model

Files changed (1) hide show

CXR_LLAVA_HF.py +1 -1

CXR_LLAVA_HF.py CHANGED Viewed

@@ -599,7 +599,7 @@ class CXRLLAVAModel(PreTrainedModel):
             streamer = TextIteratorStreamer(self.tokenizer, skip_prompt=True, skip_special_tokens=True, timeout=15)
             if np.array(image).max()>255:
-                raise Exception("WARNING. 16-bit image is not supported.")
             image = image.convert('L') # convert to grayscale
             image = np.array(image)

             streamer = TextIteratorStreamer(self.tokenizer, skip_prompt=True, skip_special_tokens=True, timeout=15)
             if np.array(image).max()>255:
+                raise Exception("16-bit image is not supported.")
             image = image.convert('L') # convert to grayscale
             image = np.array(image)