Upload app.py with huggingface_hub

app.py

@@ -24,16 +24,27 @@ def load_model():
     """Load the trained model"""
     global model
     
+    print(f"Looking for model at: {config.BEST_MODEL_PATH}")
+    print(f"Current working directory: {os.getcwd()}")
+    print(f"Files in checkpoints/: {os.listdir('checkpoints') if os.path.exists('checkpoints') else 'Directory not found'}")
+    
     if not os.path.exists(config.BEST_MODEL_PATH):
-        print(f"Warning: Model checkpoint not found at {config.BEST_MODEL_PATH}")
+        print(f"ERROR: Model checkpoint not found at {config.BEST_MODEL_PATH}")
+        print(f"Please ensure the model file exists in the checkpoints directory")
         return False
     
-    model = get_model(num_classes=config.NUM_CLASSES, device=config.DEVICE)
-    epoch, accuracy = load_checkpoint(model, None, config.BEST_MODEL_PATH)
-    model.eval()
-    
-    print(f"Model loaded from epoch {epoch + 1} with accuracy: {accuracy:.2f}%")
-    return True
+    try:
+        model = get_model(num_classes=config.NUM_CLASSES, device=config.DEVICE)
+        epoch, accuracy = load_checkpoint(model, None, config.BEST_MODEL_PATH)
+        model.eval()
+        
+        print(f"✅ Model loaded successfully from epoch {epoch + 1} with accuracy: {accuracy:.2f}%")
+        return True
+    except Exception as e:
+        print(f"ERROR loading model: {str(e)}")
+        import traceback
+        traceback.print_exc()
+        return False
 
 
 def preprocess_image(image):
@@ -157,10 +168,28 @@ def random_sample():
         return jsonify({'error': str(e)}), 500
 
 
+
+# Load model when module is imported (for Gunicorn)
+print("=" * 60)
+print("Initializing CIFAR-10 RNN Classifier")
+print("=" * 60)
+
+model_loaded = load_model()
+
+if not model_loaded:
+    print("⚠️  WARNING: Model not loaded. Application will return errors.")
+    print("Please check that checkpoints/best_model.pth exists.")
+else:
+    print("✅ Application ready to serve requests!")
+
+print("=" * 60)
+
+
 if __name__ == '__main__':
-    # Load model
-    if load_model():
-        print("Starting Flask application...")
+    # This runs only when executed directly (not with Gunicorn)
+    if model_loaded:
+        print("Starting Flask development server...")
         app.run(debug=True, host='0.0.0.0', port=5000)
     else:
         print("Failed to load model. Please train the model first using train.py")
+