Spaces:
Runtime error
Runtime error
snaramirez872
commited on
Commit
·
ff1915c
1
Parent(s):
5b6d61e
Update app.py
Browse files
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('
|
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 |
-
'
|
80 |
-
'
|
81 |
-
'
|
82 |
-
'
|
83 |
-
'
|
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
|