Update app.py

app.py

@@ -68,7 +68,7 @@ def infer_data(model, main_feeling_keyword):
             mask = infer_input['attention_mask'].to(device)
             input_id = infer_input['input_ids'].squeeze(1).to(device)
 
-            output = clsmodel(input_id, mask)
+            output = model(input_id, mask)
             result = np.argmax(output.logits, axis=1).numpy()
             result_list.extend(result)
     return result_list