fffiloni commited on
Commit
b67470b
·
verified ·
1 Parent(s): 1803938

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +4 -3
gradio_app.py CHANGED
@@ -3,12 +3,13 @@ from huggingface_hub import InferenceClient
3
  from gradio_client import Client, handle_file
4
 
5
 
6
- def infer(text_prompt, action_prompt, oauth_token: gr.OAuthToken):
7
-
 
8
 
9
  inf_client = InferenceClient(
10
  provider="fal-ai",
11
- api_key=oauth_token.token
12
  )
13
 
14
  # output is a PIL.Image object
 
3
  from gradio_client import Client, handle_file
4
 
5
 
6
+ def infer(text_prompt, action_prompt, profile: gr.OAuthProfile, oauth_token: gr.OAuthToken):
7
+ print(profile.token)
8
+ print(oauth_token.token)
9
 
10
  inf_client = InferenceClient(
11
  provider="fal-ai",
12
+ api_key=profile.token
13
  )
14
 
15
  # output is a PIL.Image object