Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
chaoyi-wu committed Aug 8, 2023
1 parent 93a8d62 commit 01f5c18
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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```.
Expand Down

0 comments on commit 01f5c18

Please sign in to comment.