Update model.py

model.py

@@ -3,7 +3,7 @@ import xml.etree.ElementTree as ET
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
-from typing import List, Dict, Any
+from typing import List, Dict, Any, Optional
 from collections import defaultdict
 from accelerate import Accelerator
 
@@ -12,6 +12,16 @@ class DynamicModel(nn.Module):
         super(DynamicModel, self).__init__()
         self.sections = nn.ModuleDict()
 
+        # Default section if none provided
+        if not sections:
+            sections = {
+                'default': [{
+                    'input_size': 128,
+                    'output_size': 256,
+                    'activation': 'relu'
+                }]
+            }
+
         for section_name, layers in sections.items():
             self.sections[section_name] = nn.ModuleList()
             for layer_params in layers:
@@ -42,9 +52,9 @@ def parse_xml_file(file_path: str) -> List[Dict[str, Any]]:
     layers = []
     for prov in root.findall('.//prov'):
         layer_params = {
-            'input_size': 128,  # Example: fixed input size
-            'output_size': 256,  # Example: fixed output size
-            'activation': 'relu'  # Default activation
+            'input_size': 128,
+            'output_size': 256,
+            'activation': 'relu'
         }
         layers.append(layer_params)
 
@@ -53,9 +63,15 @@ def parse_xml_file(file_path: str) -> List[Dict[str, Any]]:
 def create_model_from_folder(folder_path: str) -> DynamicModel:
     sections = defaultdict(list)
 
+    if not os.path.exists(folder_path):
+        print(f"Warning: Folder {folder_path} does not exist. Creating model with default configuration.")
+        return DynamicModel({})
+
+    xml_files_found = False
     for root, dirs, files in os.walk(folder_path):
         for file in files:
             if file.endswith('.xml'):
+                xml_files_found = True
                 file_path = os.path.join(root, file)
                 try:
                     layers = parse_xml_file(file_path)
@@ -64,16 +80,20 @@ def create_model_from_folder(folder_path: str) -> DynamicModel:
                 except Exception as e:
                     print(f"Error processing {file_path}: {str(e)}")
 
-    return DynamicModel(sections)
+    if not xml_files_found:
+        print("Warning: No XML files found. Creating model with default configuration.")
+        return DynamicModel({})
+
+    return DynamicModel(dict(sections))
 
 def main():
-    folder_path = 'app/data'
+    folder_path = 'Xml_Data'
     model = create_model_from_folder(folder_path)
 
     print(f"Created dynamic PyTorch model with sections: {list(model.sections.keys())}")
 
     # Get first section's first layer's input size dynamically
-    first_section = list(model.sections.keys())[0]
+    first_section = next(iter(model.sections.keys()))
     first_layer = model.sections[first_section][0]
     input_features = first_layer[0].in_features