Update app.py

app.py

@@ -26,9 +26,6 @@ model = snapshot_download(repo_id="akhaliq/frame-interpolation-film-style")
 interpolator = interpolator.Interpolator(model, None)
 
 def generate_story(choice, input_text):
-    print(choice)
-    print(input_text)
-
     query = "<BOS> <{0}> {1}".format(choice, input_text)
     
     print(query)
@@ -47,7 +44,6 @@ def generate_images(generated_text):
     image_bytes = image_gen(generated_text, steps, width, height, num_images, diversity)
     
     # Algo from spaces/Gradio-Blocks/latent_gpt2_story/blob/main/app.py
-    print(len(image_bytes))
     generated_images = []
     for image in image_bytes[1]:
         image_str = image[0]
@@ -112,6 +108,9 @@ with demo:
             output_gallery = gr.Gallery(label="Generated Story Images")
             output_interpolation = gr.Video(label="Generated Video")
             
+    with gr.Row():
+        gr.Markdown("![visitor badge](https://visitor-badge.glitch.me/badge?page_id=gradio-blocks_story_and_video_generation)")
+            
     # Bind functions to buttons
     button_gen_story.click(fn=generate_story, inputs=[input_story_type , input_start_text], outputs=output_generated_story)
     button_gen_images.click(fn=generate_images, inputs=output_generated_story, outputs=output_gallery)