Cleanup

TailUFPB · Apr 6, 2024 · 67d62cc · 67d62cc
1 parent af74d83
commit 67d62cc
Showing 1 changed file with 0 additions and 7 deletions.
diff --git a/api/DataProcesser.py b/api/DataProcesser.py
@@ -95,13 +95,6 @@ def load_weights_and_model(self, name):
  else:
  raise FileNotFoundError(f"Model file '{model_filename}' not found.")
 
- def predict_with_model(self, data, model):
- # Convert data to PyTorch tensor if needed
- data_tensor = torch.tensor(data, dtype=torch.float32)
- # Perform prediction
- predictions = model(data_tensor)
- return predictions
-
  def trained_predict(self, df, model_name):
  label_map_filename = f"api/encoders/LabelMapping-{model_name}.joblib"
  label_encoder = joblib.load(label_map_filename)