Update utils.py

utils.py

@@ -5,14 +5,14 @@ import streamlit as st
 import os
 import sys
 import argparse
-#import clip
+import clip
 import numpy as np
 from PIL import Image
-#from dalle.models import Dalle
-#from dalle.utils.utils import set_seed, clip_score
+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
+from IPython.display import display
 import random
 
 def link(link, text, **style):
@@ -113,7 +113,7 @@ def generate2(prompt,crazy,k):
         dtype=torch.float32,
         device='cpu',
         is_mega=False, 
-        is_reusable=False
+        is_reusable=True
     )
 
     # Sampling
@@ -126,9 +126,9 @@ def generate2(prompt,crazy,k):
         seed=np.random.randint(0,10000),
         grid_size=1,
         is_seamless=False,
-        temperature=1,
-        top_k=256,#2128,
-        supercondition_factor=1,
+        temperature=crazy,
+        top_k=k,#2128,
+        supercondition_factor=32,
         is_verbose=False
     )