Update utils.py

utils.py

@@ -10,6 +10,7 @@ import numpy as np
 from PIL import Image
 from dalle.models import Dalle
 from dalle.utils.utils import set_seed, clip_score
+import streamlit.components.v1 as components
 
 def link(link, text, **style):
     return a(_href=link, _target="_blank", style=styles(**style))(text)
@@ -64,6 +65,8 @@ def layout(*args):
 
     st.markdown(str(foot), unsafe_allow_html=True)
 
+
+
 def footer():
     myargs = [
         "Created by ",
@@ -78,6 +81,22 @@ def footer():
     layout(*myargs)
 
 
+    components.html(
+    """
+    <!-- Global site tag (gtag.js) - Google Analytics -->
+    <script async src="https://www.googletagmanager.com/gtag/js?id=G-SB6NJ9DQS7"></script>
+    <script>
+      window.dataLayer = window.dataLayer || [];
+      function gtag(){dataLayer.push(arguments);}
+      gtag('js', new Date());
+
+      gtag('config', 'G-SB6NJ9DQS7');
+    </script>
+    """,
+    height=600,
+    )
+
+
 model = False
 def generate(prompt,crazy,k):
     global model
@@ -93,8 +112,6 @@ def generate(prompt,crazy,k):
     
     set_seed(np.random.randint(0,10000))
 
-
-
     # Sampling
     images = model.sampling(prompt=prompt,
                             top_k=2048,