sasha HF Staff commited on
Commit
e200e49
·
verified ·
1 Parent(s): fb657c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -20,7 +20,6 @@ API = HfApi(token=TOKEN)
20
  requests= load_dataset("EnergyStarAI/requests_debug", split="test", token=TOKEN)
21
  requests_dset = requests.to_pandas()
22
  model_list= requests_dset[requests_dset['status'] == 'COMPLETED']
23
- print(model_list)
24
 
25
  tasks = ['ASR', 'Object Detection', 'Text Classification', 'Image Captioning', 'Question Answering', 'Text Generation', 'Image Classification',
26
  'Sentence Similarity', 'Image Generation', 'Summarization']
@@ -53,7 +52,7 @@ def add_new_eval(
53
  model_name = repo_id.split("/")[1]
54
 
55
  current_time = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ")
56
- if repo_id in model_list:
57
  print('This model has already been run!')
58
  else:
59
  # Is the model info correctly filled?
 
20
  requests= load_dataset("EnergyStarAI/requests_debug", split="test", token=TOKEN)
21
  requests_dset = requests.to_pandas()
22
  model_list= requests_dset[requests_dset['status'] == 'COMPLETED']
 
23
 
24
  tasks = ['ASR', 'Object Detection', 'Text Classification', 'Image Captioning', 'Question Answering', 'Text Generation', 'Image Classification',
25
  'Sentence Similarity', 'Image Generation', 'Summarization']
 
52
  model_name = repo_id.split("/")[1]
53
 
54
  current_time = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ")
55
+ if repo_id in model_list['model'].tolist():
56
  print('This model has already been run!')
57
  else:
58
  # Is the model info correctly filled?