diff --git a/README.md b/README.md index 922fb97..a148bd1 100644 --- a/README.md +++ b/README.md @@ -18,7 +18,11 @@ For quick start, you can check the `Quick_demo` path. We demonstrate a simple di - S1. Download [Model checkpoint](https://huggingface.co/chaoyi-wu/RadFM) - S2. Decompress the original zip file, you can get a `pytorch_model.bin` - S3. put `pytorch_model.bin` under path `Quick_demo/` -- S4. python `test.py` and you can get a conversation as `Input: Can you identify any visible signs of Cardiomegaly in the image? Output: yes.` +- S4. python `test.py` and you can get a conversation as + ```Input: Can you identify any visible signs of Cardiomegaly in the image? ``` + ```Output: yes.``` + +By the way, never try to perform this in cpu :)and GPU is all you need. ## Pre-train: For re-train a model on our dataset or large-sclae test our pre-train model you can check ```src```.