initial commit

app.py

@@ -60,7 +60,7 @@ class MultimodalRiskBehaviorModel(nn.Module):
             model.load_state_dict(state_dict)
         
         else:
-            hf_model = AutoModelForSequenceClassification.from_pretrained(load_directory, num_labels=1)
+            hf_model = AutoModelForSequenceClassification.from_pretrained(load_directory, num_labels=2)
             model = cls(text_model_name=hf_model.config.name_or_path, hidden_dim=hf_model.config.hidden_size)
             model.text_model = hf_model