Update tasks/text.py

tasks/text.py

@@ -112,12 +112,15 @@ async def evaluate_text(request: TextEvaluationRequest):
     
     logging.info(f"Xtrain_multi_shape:{X_train_multi.shape}")
     logging.info(f"Xtrain shape:{X_train.shape}")
+
+    logging.info(f"ytrain shape:{y_train.shape}")
+    
                   
 
     #predictions
     y_pred_bin = xgb_bin.predict(X_train)
 
-    y_pred_multi = xgb_multi.predict(X_train_multi) + 1
+    y_pred_multi = xgb_multi.predict(X_train_multi.reshape(-1,768)) + 1
 
     y_pred_bin[y_pred_bin==1] = y_pred_multi