Add demo start

app.py

@@ -0,0 +1,41 @@
+import streamlit as st # HF spaces at v1.2.0
+from demo import load_model,generate
+
+# TODOs
+# Add markdown short readme project intro
+# project setup:
+
+# git clone https://github.com/huggingface/community-events.git
+# cd community-events
+# pip install .
+ 
+st.title("ButterflyGAN")
+st.write("## This butterfly does not exist! ")
+st.write("Demo prep still in progress!!")
+
+@st.experimental_singleton
+def load_model_intocache(model_name):
+    
+    # model_name='ceyda/butterfly_512_base'
+    gan = load_model(model_name)
+
+    return gan
+
+model_name='ceyda/butterfly_cropped_uniq1K_512'
+model=load_model_intocache(model_name)
+
+st.write(f"Model {model_name} is loaded")
+st.write(f"Latent dimension: {model.latent_dim}, Image size:{model.image_size}")
+
+run=st.button("Generate")
+if run:
+    with st.spinner("Generating..."):
+    
+        batch_size=4 #generate 4 butterflies 
+        ims=generate(model,batch_size)
+
+        cols=st.columns(batch_size)
+        for i,im in enumerate(ims):
+            cols[i].image(im)
+
+

demo.py

@@ -0,0 +1,28 @@
+import torch
+from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
+from datasets import load_dataset
+
+def get_train_data(dataset_name="ceyda/smithsonian_butterflies_transparent_cropped",data_limit=1000):
+    dataset=load_dataset(dataset_name)
+    dataset=dataset.sort("sim_score")
+    score_thresh = dataset["train"][data_limit]['sim_score']
+    dataset = dataset.filter(lambda x: x['sim_score'] < score_thresh)
+    
+    dataset = dataset.map(lambda x: x.convert("RGB"))
+    return dataset["train"]
+    
+    
+
+def load_model(model_name='ceyda/butterfly_cropped_uniq1K_512'):
+    gan = LightweightGAN.from_pretrained(model_name)
+    gan.eval();
+    return gan
+    
+def generate(gan,batch_size=1):
+    with torch.no_grad():
+        ims = gan.G(torch.randn(batch_size, gan.latent_dim)).clamp_(0., 1.)
+        ims = ims.permute(0,2,3,1).detach().cpu().numpy()
+    return ims
+
+def interpolate():
+    pass

requirements.txt

@@ -0,0 +1 @@
+git+https://github.com/huggingface/community-events.git@3fea10c5d5a50c69f509e34cd580fe9139905d04#egg=huggan