mixmatch fix2

app.py

@@ -34,17 +34,22 @@ def ssl_predict(in_text, model_type):
         return_tensors="tf"
     )
 
+    preds = None
     if model_type == "fixmatch":
         model = FixMatchTune(encoder_name="readerbench/RoBERT-base")
         model.load_weights("./checkpoints/fixmatch_tune")
+        preds, _ = model([toks["input_ids"],toks["attention_mask"]], training=False)
+    
     elif model_type == "freematch":
         model = FixMatchTune(encoder_name="andrei-saceleanu/ro-offense-freematch")
         model.cls_head.load_weights("./checkpoints/freematch_tune")
+        preds, _ = model([toks["input_ids"],toks["attention_mask"]], training=False)
+    
     elif model_type == "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)
 
-    preds, _ = model([toks["input_ids"],toks["attention_mask"]], training=False)
     probs = list(preds[0].numpy())
 
     d = {}