Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -92,7 +92,7 @@ def predict_percept(image):
|
|
92 |
img = emo_mem_preprocess(img)
|
93 |
|
94 |
with torch.no_grad():
|
95 |
-
iqa_score =
|
96 |
mem_score = mem_model(img).item()
|
97 |
outputs = emotion_model(img)
|
98 |
predicted = outputs.argmax(1).item()
|
|
|
92 |
img = emo_mem_preprocess(img)
|
93 |
|
94 |
with torch.no_grad():
|
95 |
+
iqa_score = iqa_model(batch).cpu().numpy()
|
96 |
mem_score = mem_model(img).item()
|
97 |
outputs = emotion_model(img)
|
98 |
predicted = outputs.argmax(1).item()
|