A19grey commited on
Commit
5a36ad5
·
1 Parent(s): 56b0bc9

add numpy import and upate reqs

Browse files
Files changed (3) hide show
  1. __pycache__/app.cpython-38.pyc +0 -0
  2. app.py +2 -1
  3. requirements.txt +2 -1
__pycache__/app.cpython-38.pyc ADDED
Binary file (936 Bytes). View file
 
app.py CHANGED
@@ -4,6 +4,7 @@ import glob
4
  import time
5
  import random
6
  import requests
 
7
 
8
  # Import necessary libraries
9
  from torchvision import models, transforms
@@ -13,7 +14,7 @@ import torch
13
  # Load pre-trained ResNet model once
14
  model = models.resnet50(pretrained=True)
15
  model.eval()
16
-
17
  # Define image transformations
18
  transform = transforms.Compose([
19
  transforms.Resize(256),
 
4
  import time
5
  import random
6
  import requests
7
+ import numpy as np
8
 
9
  # Import necessary libraries
10
  from torchvision import models, transforms
 
14
  # Load pre-trained ResNet model once
15
  model = models.resnet50(pretrained=True)
16
  model.eval()
17
+ #
18
  # Define image transformations
19
  transform = transforms.Compose([
20
  transforms.Resize(256),
requirements.txt CHANGED
@@ -3,4 +3,5 @@ Pillow==9.5.0
3
  torch==1.13.1
4
  torchvision==0.14.1
5
  urllib3<1.27,>=1.25.4
6
- requests==2.31.0
 
 
3
  torch==1.13.1
4
  torchvision==0.14.1
5
  urllib3<1.27,>=1.25.4
6
+ requests==2.31.0
7
+ numpy==1.24.4