add icon and dep
Browse files- assets/logo.png +0 -0
- requirements.txt +6 -1
- utils.py +2 -0
assets/logo.png
ADDED
|
requirements.txt
CHANGED
|
@@ -1,3 +1,8 @@
|
|
| 1 |
streamlit
|
| 2 |
torch
|
| 3 |
-
numpy
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
streamlit
|
| 2 |
torch
|
| 3 |
+
numpy
|
| 4 |
+
git+https://github.com/huggingface/community-events.git@3fea10c5d5a50c69f509e34cd580fe9139905d04#egg=huggan
|
| 5 |
+
transformers
|
| 6 |
+
faiss-cpu
|
| 7 |
+
paddlehub
|
| 8 |
+
paddlepaddle
|
utils.py
CHANGED
|
@@ -3,6 +3,8 @@ import streamlit as st
|
|
| 3 |
import numpy as np
|
| 4 |
import torch
|
| 5 |
|
|
|
|
|
|
|
| 6 |
from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
|
| 7 |
|
| 8 |
def load_model(model_name="ceyda/butterfly_cropped_uniq1K_512", model_version=None):
|
|
|
|
| 3 |
import numpy as np
|
| 4 |
import torch
|
| 5 |
|
| 6 |
+
|
| 7 |
+
|
| 8 |
from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
|
| 9 |
|
| 10 |
def load_model(model_name="ceyda/butterfly_cropped_uniq1K_512", model_version=None):
|