Update utils.py

utils.py

@@ -11,6 +11,9 @@ from PIL import Image
 from dalle.models import Dalle
 from dalle.utils.utils import set_seed, clip_score
 import streamlit.components.v1 as components
+import torch 
+from IPython.display import display
+import random
 
 def link(link, text, **style):
     return a(_href=link, _target="_blank", style=styles(**style))(text)
@@ -21,7 +24,7 @@ def layout(*args):
     <style>
       # MainMenu {visibility: hidden;}
       footer {visibility: hidden;}
-     .stApp { bottom: 105px; }
+     .stApp { bottom: 125px; }
     </style>
     """
 
@@ -91,10 +94,42 @@ def footer():
 
       gtag('config', 'G-SB6NJ9DQS7');
     </script>
-    """,
-    height=600,
+    """
+    )
+
+
+from min_dalle import MinDalle
+
+def generate2(prompt,crazy,k):
+
+    
+    mm = MinDalle(
+        models_root='./pretrained',
+        dtype=torch.float32,
+        device='cpu',
+        is_mega=False, 
+        is_reusable=True
+    )
+
+
+    image = mm.generate_image(
+        text=prompt,
+        seed=np.random.randint(0,10000),
+        grid_size=1,
+        is_seamless=False,
+        temperature=crazy,
+        top_k=k,#2128,
+        supercondition_factor=32,
+        is_verbose=False
     )
 
+    item = {}
+    item['prompt'] = prompt
+    item['crazy'] = crazy
+    item['k'] = k
+    item['image'] = image
+    st.session_state.results.append(item)
+
 model = False
 def generate(prompt,crazy,k):
     global model
@@ -116,7 +151,7 @@ def generate(prompt,crazy,k):
         newPrompt += " architecture"
 
     images = model.sampling(prompt=newPrompt,
-                            top_k=2048,
+                            top_k=256,
                             top_p=None,
                             softmax_temperature=crazy,
                             num_candidates=num_candidates,
@@ -137,12 +172,11 @@ def generate(prompt,crazy,k):
     item = {}
     item['prompt'] = prompt
     item['crazy'] = crazy
-    item['k'] = k
+    item['k'] = 20
     item['image'] = Image.fromarray((result*255).astype(np.uint8))
     st.session_state.results.append(item)
 
 
-
 def drawGrid():
     master = {}