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()
|