Amitz244 commited on
Commit
d8e3261
·
verified ·
1 Parent(s): f55fe1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = model(batch).cpu().numpy()
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()