Upload app.py with huggingface_hub

app.py

@@ -9,7 +9,7 @@ import os
 # Add current directory to path for local imports
 sys.path.append(os.path.abspath('.'))
 
-from models.simple_nn import SimpleCNN
+from models.simple_nn import SimpleNN
 from models.transformer_imdb import TransformerClassifier
 
 # Constants
@@ -21,7 +21,8 @@ VOCAB_SIZE = 100684
 
 # 1. Helper functions to load models
 def load_image_model(model_type):
-    model = SimpleCNN().to(DEVICE)
+    # Using correct class name found in file check
+    model = SimpleNN().to(DEVICE)
     path = MNIST_PATH if 'MNIST' in model_type else FASHION_PATH
     if os.path.exists(path):
         model.load_state_dict(torch.load(path, map_location=DEVICE))