CorvaeOboro commited on
Commit
8a75545
1 Parent(s): daa864f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -8,7 +8,7 @@ import types
8
  from huggingface_hub import hf_hub_url, cached_download
9
 
10
  #TOKEN = os.environ['TOKEN']
11
- with open(cached_download(hf_hub_url('CorvaeOboro/gen_item_ring', 'gen_item_ring_stylegan2ada_20230218.pkl')), 'rb') as f:
12
  G = pickle.load(f)['G_ema']# torch.nn.Module
13
 
14
  device = torch.device("cpu")
@@ -56,10 +56,10 @@ def infer(num_images, interpolate):
56
  with demo:
57
  gr.Markdown(
58
  """
59
- # gen_item_ring
60
- ![item_ring_process_single](https://raw.githubusercontent.com/CorvaeOboro/gen_item/master/docs/ring/item_ring_process_single.jpg?raw=true "item_ring_process_single")
61
 
62
- creates ring item images from stylegan2ada model trained on synthetic dataset utilizing procgen and neural networks .
63
  more information here : [https://github.com/CorvaeOboro/gen_item](https://github.com/CorvaeOboro/gen_item).
64
  """)
65
  images_num = gr.inputs.Slider(default=6, label="Num Images", minimum=1, maximum=16, step=1)
 
8
  from huggingface_hub import hf_hub_url, cached_download
9
 
10
  #TOKEN = os.environ['TOKEN']
11
+ with open(cached_download(hf_hub_url('CorvaeOboro/gen_item_potion', 'gen_item_potion_stylegan2ada_20230430.pkl')), 'rb') as f:
12
  G = pickle.load(f)['G_ema']# torch.nn.Module
13
 
14
  device = torch.device("cpu")
 
56
  with demo:
57
  gr.Markdown(
58
  """
59
+ # gen_item_potion
60
+ ![item_potion_process_single](https://raw.githubusercontent.com/CorvaeOboro/gen_item/master/docs/potion/item_potion_process_single.jpg?raw=true "item_potion_process_single")
61
 
62
+ creates potion item images from stylegan2ada model trained on synthetic dataset utilizing procgen and neural networks .
63
  more information here : [https://github.com/CorvaeOboro/gen_item](https://github.com/CorvaeOboro/gen_item).
64
  """)
65
  images_num = gr.inputs.Slider(default=6, label="Num Images", minimum=1, maximum=16, step=1)