fix cpu

app.py

@@ -32,7 +32,7 @@ def load_model(model_path, device='cpu'):
     return model.to(device)
 
 # Get prediction
-def get_prediction(model, padded_sequences, img_x, device='cuda'):
+def get_prediction(model, padded_sequences, img_x, device='cpu'):
     malware_classes = ["Benign", "RedLine Stealer", "Downloader", "RAT", 
                        "Banking Trojan", "Snake Keylogger", "Spyware"]
 
@@ -61,10 +61,10 @@ def predict_malware(sha256_hash):
 
     # Load the model
     model_path = "model_dump/model_malware_lstm (1).pkl"
-    model = load_model(model_path, device=config.configuration["device"])
+    model = load_model(model_path, device='cpu')
 
     # Get the prediction
-    predicted_class = get_prediction(model, padded_sequences, img_x, config.configuration["device"])
+    predicted_class = get_prediction(model, padded_sequences, img_x, 'cpu')
 
     # Load the image for display
     img_x_display = Image.open(image_path)