Update app.py

Loading model from the hub

app.py

@@ -4,11 +4,12 @@ import numpy as np
 import streamlit as st
 import tensorflow as tf
 import matplotlib.pyplot as plt
+from huggingface_hub import from_pretrained_keras
 
 
 @st.cache(allow_output_mutation=True)
 def load_model():
-    generator = tf.keras.models.load_model("./models/")
+    generator = from_pretrained_keras("huggan/crypto-gan")
     return generator
 
 def generate(n_images, generator, seed=42, codings_size=100):