sasha HF Staff commited on
Commit
180d8c2
·
verified ·
1 Parent(s): f3e0901

Update app.py

Browse files

catching exceptions

Files changed (1) hide show
  1. app.py +12 -9
app.py CHANGED
@@ -43,15 +43,18 @@ def get_model_size(model_info: ModelInfo):
43
  return model_size
44
 
45
  def add_docker_eval(zip_file):
46
- API.upload_file(
47
- path_or_fileobj= zip_file,
48
- repo_id="EnergyStarAI/tested_proprietary_models",
49
- path_in_repo=zip_file,
50
- repo_type="dataset",
51
- commit_message="Adding logs via submission Space.",
52
- token= TOKEN
53
- )
54
- print('Uploaded logs to dataset!')
 
 
 
55
 
56
 
57
  def add_new_eval(
 
43
  return model_size
44
 
45
  def add_docker_eval(zip_file):
46
+ try:
47
+ API.upload_file(
48
+ path_or_fileobj= zip_file,
49
+ repo_id="EnergyStarAI/tested_proprietary_models",
50
+ path_in_repo=zip_file,
51
+ repo_type="dataset",
52
+ commit_message="Adding logs via submission Space.",
53
+ token= TOKEN
54
+ )
55
+ return 'Uploaded logs to dataset!'
56
+ except Exception:
57
+ return "Couldn't upload this file to the dataset. Try another one?"
58
 
59
 
60
  def add_new_eval(