Update app.py

app.py

@@ -1,7 +1,31 @@
 import gradio as gr
+import tensorflow as tf
+import numpy as np
+from matplotlib import cm
+from PIL import Image
+import imageio
 
-def greet(name):
-    return "Hello " + name + "!!"
+generator = tf.keras.models.load_model('dc_gan.h5')
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+def interpolate(start,end,steps,fps):
+    
+    input_vectors = np.squeeze(np.linspace(start,end,steps))
+    
+    image_vectors = np.array(generator(input_vectors))
+    
+    writer = imageio.get_writer('test.mp4', fps=fps)
+    
+    for im in image_vectors:
+        writer.append_data((im*255).astype('uint8'))
+    writer.close()
+    
+    return gr.Video(value = 'test.mp4')
+
+output_interpolation = gr.Video()
+demo = gr.Blocks()
+
+with demo:
+    btn = gr.Button("Submit")
+    btn.click(interpolate, inputs=[start,end,steps,fps], outputs=[output_interpolation])
+
+demo.launch()