fix

app.py

@@ -55,9 +55,9 @@ def model(direction, genie):
     return Image.fromarray(next_image)
 
 @spaces.GPU
-def handle_input(direction, state):
+def handle_input(direction):
     print(f"User clicked: {direction}")
-    new_image = model(direction, state)
+    new_image = genie(direction)
     return new_image
 
 @spaces.GPU
@@ -93,7 +93,9 @@ if __name__ == '__main__':
     with gr.Blocks() as demo:
         genie_instance = gr.State({'genie': genie})
         genie.device = "cuda"
-
+        with gr.Row():
+            gr.Textbox(label='HMA Demo: Select a prompt initial image from the gallery and Interact with arrow keys. \n'
+            'Note: the speed is limited due to free GPU in HF and the interface ', lines=1)
         with gr.Row():
             image_selector = gr.Dropdown(
                 choices=available_images, value=available_images[0], label="Select an Image"