snaramirez872 commited on
Commit
ff1915c
1 Parent(s): 5b6d61e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -38,7 +38,7 @@ def similar(image): # Function for Streamlit App
38
  mod = MDLS.resnet50(pretrained=True)
39
  mod.eval()
40
  mod = torch.nn.Sequential(*list(mod.children())[:-1])
41
- inFile = tarfile.open('/datasets/lfw.tar', 'r')
42
 
43
  feats = {}
44
  simScores = [] # Similarity Scores for Later
@@ -76,11 +76,11 @@ st.write("## OR")
76
 
77
  # Random Image Selector from 5 Pictures
78
  randImages = [
79
- '/datasets/random-images/img1.jpg',
80
- '/datasets/random-images/img2.jpg',
81
- '/datasets/random-images/img3.jpg',
82
- '/datasets/random-images/img4.jpg',
83
- '/datasets/random-images/img5.jpg'
84
  ]
85
 
86
  if st.button("Surprise Me!"): # Button
 
38
  mod = MDLS.resnet50(pretrained=True)
39
  mod.eval()
40
  mod = torch.nn.Sequential(*list(mod.children())[:-1])
41
+ inFile = tarfile.open('./datasets/lfw.tar', 'r')
42
 
43
  feats = {}
44
  simScores = [] # Similarity Scores for Later
 
76
 
77
  # Random Image Selector from 5 Pictures
78
  randImages = [
79
+ './datasets/random-images/img1.jpg',
80
+ './datasets/random-images/img2.jpg',
81
+ './datasets/random-images/img3.jpg',
82
+ './datasets/random-images/img4.jpg',
83
+ './datasets/random-images/img5.jpg'
84
  ]
85
 
86
  if st.button("Surprise Me!"): # Button