Update gradio_app.py

gradio_app.py

@@ -1,10 +1,11 @@
 import gradio as gr
 
-def infer(text_prompt, action_prompt):
+def infer(text_prompt, action_prompt, login):
     from huggingface_hub import InferenceClient
 
     inf_client = InferenceClient(
-        provider="fal-ai"
+        provider="fal-ai",
+        api_key=login
     )
 
     # output is a PIL.Image object
@@ -59,7 +60,7 @@ with gr.Blocks() as demo:
 
     submit_btn.click(
         fn = infer,
-        inputs = [text_prompt, action_prompt],
+        inputs = [text_prompt, action_prompt, login],
         outputs = [image_out, video_out]
     )