Final Pull request v2

.github/workflows/ruff.yml

@@ -24,4 +24,4 @@ jobs:
         run: pip install pre-commit
 
       - name: Run pre-commit hooks (on all files)
-        run: pre-commit run --all-files
+        run: pre-commit run --all-files --config .pre-commit-config-ci.yaml

exploratory_data_analysis.ipynb

@@ -27,21 +27,16 @@
     }
    ],
    "source": [
-    "import numpy as np\n",
     "import os \n",
     "import matplotlib.pyplot as plt\n",
     "import torch\n",
     "import pandas as pd \n",
-    "import pathlib\n",
     "import seaborn as sns\n",
     "import warnings\n",
-    "import scipy\n",
     "\n",
-    "from torch.utils.data import random_split, Dataset, DataLoader\n",
+    "from torch.utils.data import  Dataset, DataLoader\n",
     "from torchvision import transforms\n",
     "from PIL import Image\n",
-    "from collections import Counter\n",
-    "from matplotlib.colors import LinearSegmentedColormap\n",
     "from matplotlib.colors import ListedColormap"
    ]
   },

locust/READ.MD

@@ -0,0 +1,2 @@
+Run : locust -f locust/main.py --host=http://localhost:5100 in local
+locust -f locust/main.py --host=https://deepskin-back-808565425437.europe-west1.run.app

locust/main.py

@@ -0,0 +1,33 @@
+from locust import HttpUser, task, between
+import os
+
+class WebsiteUser(HttpUser):
+    wait_time = between(1, 3)
+
+    @task#(2)
+    def test_predict_endpoint(self):
+        image = "./.github/pictures/test_api.jpg"
+
+        with open(image, 'rb') as file:
+            files = {"image": (os.path.basename(image), file, "image/jpeg")}
+
+            data = {
+                "age": 35,
+                "sex": "male",
+                "localization": "scalp"
+            }
+
+            response = self.client.post("/predict", data=data, files=files)
+
+            if response.status_code == 200:
+                print("Prediction receive:", response.json())
+            else:
+                print(f"Prediction failed: {response.status_code} - {response.text}")
+
+    #@task(1)
+    #def test_frontend(self):
+    #    response = self.client.get("/")
+    #    if response.status_code == 200:
+    #        print("Frontend is accessible")
+    #    else:
+    #        print("Failed to access frontend:", response.status_code)

terraform/terraform.tfvars

@@ -1,5 +1,5 @@
-project_id          = "your-gcp-project-id"
+project_id          = "deepskin-451908"
 region              = "europe-west1"
-dataset_bucket_name = "your-dataset-bucket"
-model_bucket_name   = "your-model-bucket"
-code_bucket_name    = "your-code-bucket"
+dataset_bucket_name = "deepskin_dataset"
+model_bucket_name   = "trained_deepskin_model"
+code_bucket_name    = "deepskin_code"