traversaal-ai commited on
Commit
18f6aaf
1 Parent(s): 77bfe0b

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +2 -0
run.py CHANGED
@@ -1,9 +1,11 @@
1
  import gradio as gr
2
  import random
3
  import time
 
4
 
5
  endpoint_url = "https://ko60a2m26ylqgri6.us-east-1.aws.endpoints.huggingface.cloud"
6
 
 
7
 
8
  # Streaming Client
9
  client = InferenceClient(endpoint_url, token=hf_token)
 
1
  import gradio as gr
2
  import random
3
  import time
4
+ import os
5
 
6
  endpoint_url = "https://ko60a2m26ylqgri6.us-east-1.aws.endpoints.huggingface.cloud"
7
 
8
+ hf_token = os.getenv(hf_token)
9
 
10
  # Streaming Client
11
  client = InferenceClient(endpoint_url, token=hf_token)