Update gradio_app.py

gradio_app.py

@@ -3,7 +3,8 @@ from huggingface_hub import InferenceClient
 from gradio_client import Client, handle_file
 
 
-def infer(text_prompt, action_prompt):
+def infer(text_prompt, action_prompt, login):
+    print(login)
 
     inf_client = InferenceClient(
         provider="replicate",
@@ -50,7 +51,7 @@ with gr.Blocks() as demo:
 
     with gr.Column():
 
-        gr.LoginButton()
+        login = gr.LoginButton()
 
         text_prompt = gr.Textbox(label="Icon label")
         action_prompt = gr.Textbox(label="Action label")
@@ -60,7 +61,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]
     )