Merge branch 'main' into Beta-b

Aydinhamedi · Feb 29, 2024 · 9686df3 · 9686df3
2 parents b92a916 + 7c4dc70
commit 9686df3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 9 deletions.
diff --git a/README.md b/README.md
@@ -26,6 +26,14 @@
 > ```python
 > import efficientnet.tfkeras
 > ```
+> Or clone the project and use `Utils.FixedDropout`:
+> ```python
+> from Utils.FixedDropout import FixedDropout
+> from keras.models import load_model
+> 
+> # Load the model
+> model = load_model('PAI_model_T.h5', custom_objects={'FixedDropout': FixedDropout})
+> ```
 
 The project includes a Command Line Interface (CLI) and a (GUI) Graphical User Interface for easy use of the model. The CLI, which is based on the [Python CLI template](https://github.com/Aydinhamedi/Python-CLI-template) from the same author, provides a user-friendly, colorful interface that allows you to interact with the model. you can fined the CLI in 
 
@@ -40,14 +48,7 @@ Interface\GUI
 ![Example](doc/Other/CLI_V0.8.9.3.png)  
 ### Example Image of the GUI (V0.8.9.6) ⤵
 ![Example](doc/Other/GUI_V0.8.9.6.png)  
-#### __If you want to use the model in your project:__
-```python
-from Utils.FixedDropout import FixedDropout
-from keras.models import load_model
 
-# Load the model
-model = load_model('PAI_model_T.h5', custom_objects={'FixedDropout': FixedDropout})
-```
 ## Release
 > ### Newest release 📃
 > #### [Go to newest release](https://github.com/Aydinhamedi/Pneumonia-Detection-Ai/releases/latest)
@@ -72,7 +73,7 @@ This combined dataset provides a comprehensive set of images for training the mo
 ### Model list:
 | Model    | Base Model      | Params | acc    | Status |
 |----------|-----------------|--------|--------|--------|
-| V6       | efficientnet-b7 | 65.4M  | 97.12% |   ⚠️   |
+| V6       | efficientnet-b7 | 65.4M  | 97.12% |   ✅   |
 | V7 light | efficientnet-b4 | 29.7M  | 97.12% |   ✅  |
 | V8 Super light | efficientnet-b0 | 4.8M  | 96.47% |   ✅  |
 

diff --git a/SECURITY.md b/SECURITY.md
@@ -3,7 +3,7 @@
 ## Model list:
 | Model    | Base Model      | Params | acc    | Status |
 |----------|-----------------|--------|--------|--------|
-| V6       | efficientnet-b7 | 65.4M  | 97.12% |   ⚠️   |
+| V6       | efficientnet-b7 | 65.4M  | 97.12% |   ✅   |
 | V7 light | efficientnet-b4 | 29.7M  | 97.12% |   ✅  |
 | V8 Super light | efficientnet-b0 | 4.8M  | 96.47% |   ✅  |