Spaces:
Sleeping
Sleeping
CorvaeOboro
commited on
Commit
•
8a75545
1
Parent(s):
daa864f
Update app.py
Browse files
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/
|
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 |
-
#
|
60 |
-
![
|
61 |
|
62 |
-
creates
|
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)
|