Update app.py

app.py

@@ -104,19 +104,6 @@ def selectall(text):
         doclist += r
     return doclist 
 
-# image generator
-image_gen = gr.Interface.load("spaces/multimodalart/latentdiffusion")
-
-# video generator
-os.system("git clone https://github.com/google-research/frame-interpolation")
-sys.path.append("frame-interpolation")
-from eval import interpolator, util
-
-ffmpeg_path = util.get_ffmpeg_path()
-mediapy.set_ffmpeg(ffmpeg_path)
-model = snapshot_download(repo_id="akhaliq/frame-interpolation-film-style")
-interpolator = interpolator.Interpolator(model, None)
-
 # story gen
 def generate_story(choice, input_text):
     query = "<BOS> <{0}> {1}".format(choice, input_text)
@@ -160,8 +147,21 @@ def generate_interpolation(gallery):
     mediapy.write_video("out.mp4", frames, fps=15)
     return "out.mp4"
 
-demo = gr.Blocks()
 
+# image generator
+image_gen = gr.Interface.load("spaces/multimodalart/latentdiffusion")
+
+# video generator
+os.system("git clone https://github.com/google-research/frame-interpolation")
+sys.path.append("frame-interpolation")
+from eval import interpolator, util
+
+ffmpeg_path = util.get_ffmpeg_path()
+mediapy.set_ffmpeg(ffmpeg_path)
+model = snapshot_download(repo_id="akhaliq/frame-interpolation-film-style")
+interpolator = interpolator.Interpolator(model, None)
+
+demo = gr.Blocks()
 with demo:
 
     audio_file = gr.inputs.Audio(source="microphone", type="filepath")
@@ -169,7 +169,7 @@ with demo:
     label = gr.Label()
     saved = gr.Textbox()
     savedAll = gr.Textbox()
-    TTSchoice = gr.inputs.Radio( label="Pick a TTS Model", choices=MODEL_NAMES,   )
+    TTSchoice = gr.inputs.Radio( label="Pick a TTS Model", choices=MODEL_NAMES  )
     audio = gr.Audio(label="Output", interactive=False)
     
     b1 = gr.Button("Recognize Speech")
@@ -184,43 +184,17 @@ with demo:
     b4.click(selectall, inputs=text, outputs=savedAll)
     b5.click(tts,  inputs=[text,TTSchoice], outputs=audio)
     
-            
-    with gr.Row():
-        # Left column (inputs)
-        with gr.Column():
-            input_story_type = gr.Radio(choices=['superhero', 'action', 'drama', 'horror', 'thriller', 'sci_fi'], value='sci_fi', label="Genre")
-            input_start_text = gr.Textbox(placeholder='A teddy bear outer space', label="Starting Text")
-            
-            gr.Markdown("Be sure to run each of the buttons one at a time, they depend on each others' outputs!")
-            
-            # Rows of instructions & buttons
-            with gr.Row():
-                gr.Markdown("1. Select a type of story, then write some starting text! Then hit the 'Generate Story' button to generate a story! Feel free to edit the generated story afterwards!")
-                button_gen_story = gr.Button("Generate Story")
-            with gr.Row():
-                gr.Markdown("2. After generating a story, hit the 'Generate Images' button to create some visuals for your story! (Can re-run multiple times!)")
-                button_gen_images = gr.Button("Generate Images")
-            with gr.Row():
-                gr.Markdown("3. After generating some images, hit the 'Generate Video' button to create a short video by interpolating the previously generated visuals!")
-                button_gen_video = gr.Button("Generate Video")
-                
-            # Rows of references
-            with gr.Row():
-                gr.Markdown("--Models Used--")
-            with gr.Row():
-                gr.Markdown("Story Generation: [GPT-J](https://huggingface.co/pranavpsv/gpt2-genre-story-generator)")
-            with gr.Row():
-                gr.Markdown("Image Generation Conditioned on Text: [Latent Diffusion](https://huggingface.co/spaces/multimodalart/latentdiffusion) | [Github Repo](https://github.com/CompVis/latent-diffusion)")
-            with gr.Row():
-                gr.Markdown("Interpolations: [FILM](https://huggingface.co/spaces/akhaliq/frame-interpolation) | [Github Repo](https://github.com/google-research/frame-interpolation)")
-            with gr.Row():
-                gr.Markdown("![visitor badge](https://visitor-badge.glitch.me/badge?page_id=gradio-blocks_story_and_video_generation)")
-                
-        # Right column (outputs)
-        with gr.Column():
-            output_generated_story = gr.Textbox(label="Generated Story")
-            output_gallery = gr.Gallery(label="Generated Story Images")
-            output_interpolation = gr.Video(label="Generated Video")
+    input_story_type = gr.Radio(choices=['superhero', 'action', 'drama', 'horror', 'thriller', 'sci_fi'], value='sci_fi', label="Genre")
+    input_start_text = gr.Textbox(placeholder='A teddy bear outer space', label="Starting Text")
+    gr.Markdown("1. Select a type of story, then write some starting text! Then hit the 'Generate Story' button to generate a story! Feel free to edit the generated story afterwards!")
+    button_gen_story = gr.Button("Generate Story")
+    gr.Markdown("2. After generating a story, hit the 'Generate Images' button to create some visuals for your story! (Can re-run multiple times!)")
+    button_gen_images = gr.Button("Generate Images")
+    gr.Markdown("3. After generating some images, hit the 'Generate Video' button to create a short video by interpolating the previously generated visuals!")
+    button_gen_video = gr.Button("Generate Video")
+    output_generated_story = gr.Textbox(label="Generated Story")
+    output_gallery = gr.Gallery(label="Generated Story Images")
+    output_interpolation = gr.Video(label="Generated Video")
             
     # Bind functions to buttons
     button_gen_story.click(fn=generate_story, inputs=[input_story_type , input_start_text], outputs=output_generated_story)