mixmatch fix

app.py

@@ -41,7 +41,7 @@ def ssl_predict(in_text, model_type):
         model = FixMatchTune(encoder_name="andrei-saceleanu/ro-offense-freematch")
         model.cls_head.load_weights("./checkpoints/freematch_tune")
     elif model_type == "mixmatch":
-        model = MixMatch(encoder_name="andrei-saceleanu/ro-offense-mixmatch")
+        model = MixMatch(bert_model="andrei-saceleanu/ro-offense-mixmatch")
         model.cls_head.load_weights("./checkpoints/mixmatch")
 
     preds, _ = model([toks["input_ids"],toks["attention_mask"]], training=False)