Spaces:
Sleeping
Sleeping
dcavadia commited on
Commit ·
9bdbbe5
1
Parent(s): bc1fb7d
update model info
Browse files- src/core/model.py +22 -0
src/core/model.py
CHANGED
|
@@ -137,3 +137,25 @@ class MelanoScopeModel:
|
|
| 137 |
except:
|
| 138 |
empty_df = None
|
| 139 |
return (message, "", "", "", "", "", empty_df, "")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 137 |
except:
|
| 138 |
empty_df = None
|
| 139 |
return (message, "", "", "", "", "", empty_df, "")
|
| 140 |
+
|
| 141 |
+
def get_model_info(self) -> Dict[str, Any]:
|
| 142 |
+
"""Get model information and metadata."""
|
| 143 |
+
info = {
|
| 144 |
+
"classes": self.classes,
|
| 145 |
+
"num_classes": len(self.classes),
|
| 146 |
+
"model_file": str(MODEL_FILE),
|
| 147 |
+
"providers": ModelConfig.ORT_PROVIDERS
|
| 148 |
+
}
|
| 149 |
+
|
| 150 |
+
if self.session:
|
| 151 |
+
try:
|
| 152 |
+
input_info = self.session.get_inputs()[0]
|
| 153 |
+
info.update({
|
| 154 |
+
"input_shape": input_info.shape,
|
| 155 |
+
"input_type": input_info.type,
|
| 156 |
+
"input_name": input_info.name
|
| 157 |
+
})
|
| 158 |
+
except Exception as e:
|
| 159 |
+
logger.warning(f"Could not get model input info: {e}")
|
| 160 |
+
|
| 161 |
+
return info
|