Update app.py

app.py

@@ -1,3 +1,12 @@
+from huggingface_hub import from_pretrained_keras
+import matplotlib.pyplot as plt
+from math import sqrt, ceil
+import tensorflow as tf
+import gradio as gr
+import numpy as np
+
+model = from_pretrained_keras("IMvision12/WGAN-GP")
+
 def create_digit_samples(num_images):
       random_latent_vectors = tf.random.normal(shape=(int(num_images), 128))
       predictions = model.predict(random_latent_vectors)