Spaces:
Runtime error
Runtime error
removed print statements
Browse files- bayes/models.py +0 -1
bayes/models.py
CHANGED
@@ -77,7 +77,6 @@ def process_imagenet_get_model(data):
|
|
77 |
for q in range(0, perturbed_images.shape[0], batch_size):
|
78 |
predictions.append(softmax(model(perturbed_images[q:q+batch_size])).cpu().detach().numpy())
|
79 |
predictions = np.concatenate(predictions, axis=0)
|
80 |
-
print("Predictions made in models.py:", predictions)
|
81 |
return predictions
|
82 |
return wrapped_model
|
83 |
|
|
|
77 |
for q in range(0, perturbed_images.shape[0], batch_size):
|
78 |
predictions.append(softmax(model(perturbed_images[q:q+batch_size])).cpu().detach().numpy())
|
79 |
predictions = np.concatenate(predictions, axis=0)
|
|
|
80 |
return predictions
|
81 |
return wrapped_model
|
82 |
|