Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	
		Clémentine
		
	commited on
		
		
					Commit 
							
							·
						
						69297e0
	
1
								Parent(s):
							
							7fe34a6
								
fixes display for model params that are 0
Browse files
    	
        src/display_models/get_model_metadata.py
    CHANGED
    
    | 
         @@ -22,8 +22,8 @@ def get_model_infos_from_hub(leaderboard_data: List[dict]): 
     | 
|
| 22 | 
         
             
                        model_info = api.model_info(model_name)
         
     | 
| 23 | 
         
             
                    except huggingface_hub.utils._errors.RepositoryNotFoundError:
         
     | 
| 24 | 
         
             
                        print("Repo not found!", model_name)
         
     | 
| 25 | 
         
            -
                        model_data[AutoEvalColumn.license.name] =  
     | 
| 26 | 
         
            -
                        model_data[AutoEvalColumn.likes.name] =  
     | 
| 27 | 
         
             
                        model_data[AutoEvalColumn.params.name] = get_model_size(model_name, None)
         
     | 
| 28 | 
         
             
                        continue
         
     | 
| 29 | 
         | 
| 
         @@ -36,7 +36,7 @@ def get_model_license(model_info): 
     | 
|
| 36 | 
         
             
                try:
         
     | 
| 37 | 
         
             
                    return model_info.cardData["license"]
         
     | 
| 38 | 
         
             
                except Exception:
         
     | 
| 39 | 
         
            -
                    return  
     | 
| 40 | 
         | 
| 41 | 
         | 
| 42 | 
         
             
            def get_model_likes(model_info):
         
     | 
| 
         @@ -56,7 +56,7 @@ def get_model_size(model_name, model_info): 
     | 
|
| 56 | 
         
             
                        size = size_match.group(0)
         
     | 
| 57 | 
         
             
                        return round(float(size[:-1]) if size[-1] == "b" else float(size[:-1]) / 1e3, 3)
         
     | 
| 58 | 
         
             
                    except AttributeError:
         
     | 
| 59 | 
         
            -
                        return  
     | 
| 60 | 
         | 
| 61 | 
         | 
| 62 | 
         
             
            def get_model_type(leaderboard_data: List[dict]):
         
     | 
| 
         | 
|
| 22 | 
         
             
                        model_info = api.model_info(model_name)
         
     | 
| 23 | 
         
             
                    except huggingface_hub.utils._errors.RepositoryNotFoundError:
         
     | 
| 24 | 
         
             
                        print("Repo not found!", model_name)
         
     | 
| 25 | 
         
            +
                        model_data[AutoEvalColumn.license.name] = "?"
         
     | 
| 26 | 
         
            +
                        model_data[AutoEvalColumn.likes.name] = 0
         
     | 
| 27 | 
         
             
                        model_data[AutoEvalColumn.params.name] = get_model_size(model_name, None)
         
     | 
| 28 | 
         
             
                        continue
         
     | 
| 29 | 
         | 
| 
         | 
|
| 36 | 
         
             
                try:
         
     | 
| 37 | 
         
             
                    return model_info.cardData["license"]
         
     | 
| 38 | 
         
             
                except Exception:
         
     | 
| 39 | 
         
            +
                    return "?"
         
     | 
| 40 | 
         | 
| 41 | 
         | 
| 42 | 
         
             
            def get_model_likes(model_info):
         
     | 
| 
         | 
|
| 56 | 
         
             
                        size = size_match.group(0)
         
     | 
| 57 | 
         
             
                        return round(float(size[:-1]) if size[-1] == "b" else float(size[:-1]) / 1e3, 3)
         
     | 
| 58 | 
         
             
                    except AttributeError:
         
     | 
| 59 | 
         
            +
                        return 0
         
     | 
| 60 | 
         | 
| 61 | 
         | 
| 62 | 
         
             
            def get_model_type(leaderboard_data: List[dict]):
         
     | 
    	
        src/display_models/read_results.py
    CHANGED
    
    | 
         @@ -27,7 +27,7 @@ class EvalResult: 
     | 
|
| 27 | 
         
             
                results: dict
         
     | 
| 28 | 
         
             
                precision: str = ""
         
     | 
| 29 | 
         
             
                model_type: str = ""
         
     | 
| 30 | 
         
            -
                weight_type: str = ""
         
     | 
| 31 | 
         
             
                date: str = ""
         
     | 
| 32 | 
         | 
| 33 | 
         
             
                def to_dict(self):
         
     | 
| 
         | 
|
| 27 | 
         
             
                results: dict
         
     | 
| 28 | 
         
             
                precision: str = ""
         
     | 
| 29 | 
         
             
                model_type: str = ""
         
     | 
| 30 | 
         
            +
                weight_type: str = "Original"
         
     | 
| 31 | 
         
             
                date: str = ""
         
     | 
| 32 | 
         | 
| 33 | 
         
             
                def to_dict(self):
         
     |