Update test.py

test.py

@@ -59,13 +59,3 @@ def predict(model, embeddings):
         
         return {"category_scores": result, 'detect': detected, 'detected': detect_value}
 
-
-print('Load model')
-moderation = ModerationModel()
-moderation.load_state_dict(torch.load('moderation_model.pth'))
-
-text = "I want to kill them."
-
-embeddings_for_prediction = getEmb(text)
-prediction = predict(moderation, embeddings_for_prediction)
-print(json.dumps(prediction,indent=4))