kadirnar
/

bsrgan

kadirnar commited on Dec 22, 2022

Commit

e80840d

1 Parent(s): 7217532

Update README.md

Files changed (1) hide show

README.md CHANGED Viewed

@@ -23,7 +23,7 @@ pip install bsrgan
 ```python
 from bsrgan import BSRGAN
-model = BSRGAN(weights='kadirnar/bsrgan', device='cuda:0')
 model.save = True
 pred = model.predict(img_path='data/image/test.png')

 ```python
 from bsrgan import BSRGAN
+model = BSRGAN(weights='kadirnar/bsrgan', device='cuda:0', hf_model=True)
 model.save = True
 pred = model.predict(img_path='data/image/test.png')