Update app.py

app.py

@@ -1,15 +1,3 @@
-# -*- coding: utf-8 -*-
-"""20221027_Generate_Beatles_with_Gradio_faster_version (2).ipynb
-
-Automatically generated by Colaboratory.
-
-Original file is located at
-    https://colab.research.google.com/drive/1B2KVLt04uBMgr2ejW19FYaEgk9lnNS0E
-
-# Build a demo
-
-Put together a Gradio App bases on the Huggingface tutorials so that users can generate Beatles-like poetry based on an input prompt.
-"""
 # Import libraries
 from transformers import pipeline
 from numpy import random
@@ -46,8 +34,9 @@ checkpoint = 'wvangils/BLOOM-560m-Beatles-Lyrics-finetuned'
 # Setup all the pipelines we need
 title_generator = pipeline('summarization', model='czearing/story-to-title')
 lyrics_generator = pipeline("text-generation", model=checkpoint)
-# For the image generator we use stable diffusion from an existing HuggingFace space
-stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion-2")
+
+# For the image generator we use stable diffusion from an existing HuggingFace space, Gradio accelerated backend
+stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion-1")
 
 # Create 4 images for the given prompt and receive the first one
 # This function uses an existing HuggingFace space where the number of created images cannot be modified
@@ -63,14 +52,10 @@ def generate_beatles(input_prompt, temperature, top_p, given_input_style):
                               , max_length = 100
                               , num_return_sequences = 1
                               , return_full_text = True
-                              #, num_beams = 1
-                              #, early_stopping = True    # Werkt niet goed lijkt 
                               , temperature = temperature    
-                              #, top_k = 50     # Default 50
                               , top_p = top_p    # Default 1.0
                               , no_repeat_ngram_size = 3   # Default = 0
                               , repetition_penalty = 1.0   # Default = 1.0
-                              #, do_sample = True     # Default = False
                               )[0]["generated_text"]
 
     # Put lyrics in the right form
@@ -118,7 +103,6 @@ css = """
 temperature = gr.Slider(minimum=0.1, maximum=1.0, step=0.1, label="Change the temperature \r\n (higher temperature = more creative in lyrics generation, but posibbly less Beatly)", value=0.7, show_label=True)  #high = sensitive for low probability tokens
 top_p = gr.Slider(minimum=0.1, maximum=1.0, step=0.1, label="Change top probability of the next word \n (higher top probability = more words to choose from for the next word, but possibly less Beatly)", value=0.5, show_label=True)
 given_input_style = gr.Dropdown(choices=image_input_styles, value="Random", label="Choose the art style for the lyrics cover", show_label=True)
-#checkpoint = gr.Radio(checkpoint_choices, value='wvangils/GPT-Medium-Beatles-Lyrics-finetuned-newlyrics', interactive=True, label = 'Select fine-tuned model', show_label=True)
 
 # Use generate Beatles function in demo-app Gradio
 gr.Interface(fn=generate_beatles