dcavadia commited on
Commit
9bdbbe5
·
1 Parent(s): bc1fb7d

update model info

Browse files
Files changed (1) hide show
  1. 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