Updated requirements.txt

Signed-off-by: Kieran Fraser <[email protected]>

app.py

@@ -171,7 +171,7 @@ def clf_poison_evaluate(*args):
         optimizer = torch.optim.Adam(model.parameters(), lr=1e-4)
         loss_fn = torch.nn.CrossEntropyLoss()
 
-        hf_model = HuggingFaceClassifierPyTorch(
+        poison_hf_model = HuggingFaceClassifierPyTorch(
             model=model,
             loss=loss_fn,
             optimizer=optimizer,
@@ -249,12 +249,12 @@ def clf_poison_evaluate(*args):
             is_poison[i] = True
 
         poison_indices = np.where(is_poison)[0]
-        hf_model.fit(x_poison, y_poison, nb_epochs=2)
+        poison_hf_model.fit(x_poison, y_poison, nb_epochs=2)
         
         clean_x = x_poison[~is_poison]
         clean_y = y_poison[~is_poison]
 
-        outputs = hf_model.predict(clean_x)
+        outputs = poison_hf_model.predict(clean_x)
         clean_preds = np.argmax(outputs, axis=1)
         clean_acc = np.mean(clean_preds == clean_y)
         
@@ -265,7 +265,7 @@ def clf_poison_evaluate(*args):
         poison_x = x_poison[is_poison]
         poison_y = y_poison[is_poison]
 
-        outputs = hf_model.predict(poison_x)
+        outputs = poison_hf_model.predict(poison_x)
         poison_preds = np.argmax(outputs, axis=1)
         poison_acc = np.mean(poison_preds == poison_y)
         

requirements.txt

@@ -1,8 +1,8 @@
 pandas
 jupyter
 transformers
-torch==1.13.1
-torchvision==0.14.0
+torch==1.13
+torchvision==0.14
 tensorflow==2.10.1; sys_platform != "darwin"
 tensorflow-macos; sys_platform == "darwin"
 tensorflow-metal; sys_platform == "darwin"