Update app.py

app.py

@@ -8,6 +8,8 @@ import numpy as np
 from datetime import datetime
 import sqlite3
 import torch.nn as nn
+import torchvision.models as models
+
 app = Flask(__name__)
 
 # ✅ Directory and database path