Soumik555 commited on
Commit
672fb1a
·
1 Parent(s): 1cf7f06

db api online

Browse files
Files changed (1) hide show
  1. model_service_api.py +3 -2
model_service_api.py CHANGED
@@ -279,12 +279,13 @@ async def get_free_model_names(
279
  try:
280
  free_models = await get_free_models_data(category=category)
281
 
282
- return free_models
283
-
284
  model_names = [
285
  {
286
  "id": model["id"],
287
  "name": model["name"],
 
 
 
288
  "description": model.get("description", "")[:100] + "..." if len(model.get("description", "")) > 100 else model.get("description", "")
289
  }
290
  for model in free_models
 
279
  try:
280
  free_models = await get_free_models_data(category=category)
281
 
 
 
282
  model_names = [
283
  {
284
  "id": model["id"],
285
  "name": model["name"],
286
+ "input_modalities": model.get("architecture", {}).get("input_modalities"),
287
+ "output_modalities": model.get("architecture", {}).get("output_modalities"),
288
+ "per_request_limits": model.get("per_request_limits", {}),
289
  "description": model.get("description", "")[:100] + "..." if len(model.get("description", "")) > 100 else model.get("description", "")
290
  }
291
  for model in free_models