Update README.md

#1
by bomze - opened
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -17,16 +17,18 @@ tags:
17
  ### Installation
18
  ```
19
  pip install bsrgan
 
20
  ```
21
 
22
  ### BSRGAN Usage
23
  ```python
24
  from bsrgan import BSRGAN
25
 
26
- model = BSRGAN(weights='kadirnar/bsrgan', device='cuda:0', hf_model=True)
27
  model.save = True
28
 
29
  pred = model.predict(img_path='data/image/test.png')
 
30
  ```
31
 
32
  ### BibTeX Entry and Citation Info
 
17
  ### Installation
18
  ```
19
  pip install bsrgan
20
+ pip install huggingface_hub
21
  ```
22
 
23
  ### BSRGAN Usage
24
  ```python
25
  from bsrgan import BSRGAN
26
 
27
+ model = BSRGAN(model_path='kadirnar/bsrgan', device='cuda:0', hf_model=True)
28
  model.save = True
29
 
30
  pred = model.predict(img_path='data/image/test.png')
31
+ print(f'Result here: {pred}')
32
  ```
33
 
34
  ### BibTeX Entry and Citation Info