diff --git a/README.md b/README.md index 12bac14..38304e1 100644 --- a/README.md +++ b/README.md @@ -21,7 +21,8 @@ Feel free to modify it as you want. - 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? + + > 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 :).