| 106 | parser.set_defaults(_command=ModelCardCMD) |
| 107 | |
| 108 | def create_model(self): |
| 109 | from modelscope.hub.constants import Licenses, ModelVisibility |
| 110 | visibilities = [ |
| 111 | getattr(ModelVisibility, attr) for attr in dir(ModelVisibility) |
| 112 | if not attr.startswith('__') |
| 113 | ] |
| 114 | if self.args.visibility not in visibilities: |
| 115 | raise ValueError('The access_token must in %s!' % visibilities) |
| 116 | licenses = [ |
| 117 | getattr(Licenses, attr) for attr in dir(Licenses) |
| 118 | if not attr.startswith('__') |
| 119 | ] |
| 120 | if self.args.license not in licenses: |
| 121 | raise ValueError('The license must in %s!' % licenses) |
| 122 | try: |
| 123 | self.api.get_model(self.model_id) |
| 124 | except Exception as e: |
| 125 | logger.info('>>> %s' % type(e)) |
| 126 | self.api.create_model( |
| 127 | model_id=self.model_id, |
| 128 | visibility=self.args.visibility, |
| 129 | license=self.args.license, |
| 130 | chinese_name=self.args.chinese_name, |
| 131 | ) |
| 132 | self.pprint() |
| 133 | |
| 134 | def get_model_url(self): |
| 135 | return self.api.get_model_url(self.model_id) |