svjack commited on
Commit
b8ff652
1 Parent(s): 681ad9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -16,13 +16,14 @@ import pathlib
16
 
17
  import imagehash
18
 
19
- MAX_MODEL_NUM = 300
20
 
 
21
  '''
22
  Yntec digplay
23
  '''
24
 
25
- hf_civital_image_info_dataset = load_dataset("svjack/hf_civital_image_info")
26
  hf_civital_image_info_df = hf_civital_image_info_dataset["train"].to_pandas()
27
 
28
  def gen_interface(model_name, max_times = 3):
 
16
 
17
  import imagehash
18
 
19
+ API_TOKEN = os.environ.get("HF_READ_TOKEN")
20
 
21
+ MAX_MODEL_NUM = 300
22
  '''
23
  Yntec digplay
24
  '''
25
 
26
+ hf_civital_image_info_dataset = load_dataset("svjack/hf_civital_image_info", token = API_TOKEN)
27
  hf_civital_image_info_df = hf_civital_image_info_dataset["train"].to_pandas()
28
 
29
  def gen_interface(model_name, max_times = 3):