Spaces:
Runtime error
Runtime error
Commit ·
1bcc710
1
Parent(s): 857919b
Changing init to use params
Browse files- main/api.py +5 -1
main/api.py
CHANGED
|
@@ -114,6 +114,10 @@ class InferenceApi(LitAPI):
|
|
| 114 |
self.logger.error(f"Error in download_model: {str(e)}")
|
| 115 |
raise
|
| 116 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 117 |
async def validate_system(self) -> Dict[str, Any]:
|
| 118 |
"""Validate system configuration and setup."""
|
| 119 |
self.logger.debug("Validating system configuration...")
|
|
@@ -138,7 +142,7 @@ class InferenceApi(LitAPI):
|
|
| 138 |
async with await self._get_client() as client:
|
| 139 |
response = await client.post(
|
| 140 |
self._get_endpoint('model_initialize'),
|
| 141 |
-
|
| 142 |
)
|
| 143 |
response.raise_for_status()
|
| 144 |
return response.json()
|
|
|
|
| 114 |
self.logger.error(f"Error in download_model: {str(e)}")
|
| 115 |
raise
|
| 116 |
|
| 117 |
+
except Exception as e:
|
| 118 |
+
self.logger.error(f"Error initiating model download: {str(e)}")
|
| 119 |
+
raise
|
| 120 |
+
|
| 121 |
async def validate_system(self) -> Dict[str, Any]:
|
| 122 |
"""Validate system configuration and setup."""
|
| 123 |
self.logger.debug("Validating system configuration...")
|
|
|
|
| 142 |
async with await self._get_client() as client:
|
| 143 |
response = await client.post(
|
| 144 |
self._get_endpoint('model_initialize'),
|
| 145 |
+
params={"model_name": model_name} if model_name else None
|
| 146 |
)
|
| 147 |
response.raise_for_status()
|
| 148 |
return response.json()
|