Spaces:
Running
on
Zero
Running
on
Zero
File size: 2,634 Bytes
48cef3d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 |
from app.models import *
# # has named endpoint
# if '/' == HF_SPACES[model]['function'][0]:
# # audio sync function name
# api_name = HF_SPACES[model]['function']
# end_parameters = _get_param_examples(
# endpoints['named_endpoints'][api_name]['parameters']
# )
# # has unnamed endpoint
# else:
# # endpoint index is the first character
# fn_index = int(HF_SPACES[model]['function'])
# end_parameters = _get_param_examples(
# endpoints['unnamed_endpoints'][str(fn_index)]['parameters']
# )
def _get_param_examples(parameters):
# named or unnamed parameters
try:
param_name = parameters[0]['parameter_name']
# success => named params, use dict
example_inputs = {}
except:
# unnamed params, use list
example_inputs = []
pass
for param_info in parameters:
param_name = ''
param_default_value = param_info['example_input']
try:
# named params
param_name = param_info['parameter_name']
param_default_value = param_info['parameter_default']
except:
# unnamed params
pass
param_value = None
if (
param_info['component'] == 'Radio'
or param_info['component'] == 'Dropdown'
or param_info['component'] == 'Audio'
or param_info['python_type']['type'] == 'str'
):
param_value = str(param_default_value)
elif param_info['python_type']['type'] == 'int':
param_value = int(param_default_value)
elif param_info['python_type']['type'] == 'float':
param_value = float(param_default_value)
elif param_info['python_type']['type'] == 'bool':
param_value = bool(param_default_value)
if (param_name != ''):
# named param
example_inputs[param_info['parameter_name']] = param_value
else:
# just append unnamed param and hope
example_inputs.append(param_value)
return example_inputs
def _override_params(inputs, modelname):
try:
for key,value in OVERRIDE_INPUTS[modelname].items():
# if override keys are integers, make the dict into a list
if (
(type(inputs) is dict)
and (type(key) is int)
):
print("Converting unnamed override params to List")
inputs = list(inputs.values())
inputs[key] = value
print(f"{modelname}: Default inputs overridden by Arena")
except:
pass
return inputs |