dhruv2842 commited on
Commit
cf2ad62
·
verified ·
1 Parent(s): c539386

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -8,6 +8,8 @@ import numpy as np
8
  from datetime import datetime
9
  import sqlite3
10
  import torch.nn as nn
 
 
11
  app = Flask(__name__)
12
 
13
  # ✅ Directory and database path
 
8
  from datetime import datetime
9
  import sqlite3
10
  import torch.nn as nn
11
+ import torchvision.models as models
12
+
13
  app = Flask(__name__)
14
 
15
  # ✅ Directory and database path