LuisV
commited on
Commit
•
4bd01c5
1
Parent(s):
d2316ea
fixing path problem
Browse files
imageprocessing/imageprocessingtools.py
CHANGED
|
@@ -22,7 +22,12 @@ clip_model, clip_preprocess = clip.load('ViT-B/32', clip_device)
|
|
| 22 |
|
| 23 |
emo_device = "cpu"
|
| 24 |
img2emo_model = torch.load(
|
| 25 |
-
"img2emo.pt",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
map_location=emo_device
|
| 27 |
)
|
| 28 |
|
|
|
|
| 22 |
|
| 23 |
emo_device = "cpu"
|
| 24 |
img2emo_model = torch.load(
|
| 25 |
+
#"img2emo.pt",
|
| 26 |
+
os.path.join(
|
| 27 |
+
os.getcwd(),
|
| 28 |
+
"imageprocessing",
|
| 29 |
+
"img2emo.pt"
|
| 30 |
+
),
|
| 31 |
map_location=emo_device
|
| 32 |
)
|
| 33 |
|