app updated

app.py

@@ -40,13 +40,15 @@ def get_prompt(prompt):
 
     return "Images generated"
 
+
 if __name__ == '__main__':
     with gr.Blocks() as demo:
         gallery = gr.Gallery(
             label="Generated images", show_label=False, elem_id="gallery",
             columns=[6], rows=[4], object_fit="contain", height="auto")  # set rows to 24 to accommodate all images
         btn = gr.Button("Generate images", scale=0)
+        text_input = gr.Interface(fn=get_prompt, inputs="text")  # create a text input interface
 
-        btn.click(get_prompt, 'text', gallery)
+        btn.click(get_prompt, text_input, gallery)  # pass the text input interface to btn.click()
 
-    demo.launch()
+    demo.launch()

requirements.txt

@@ -4,3 +4,4 @@ gradio
 torch
 torchvision
 torchtext
+accelerate