Add ChatGPT (#2341)

* initial scafold

Signed-off-by: Ashwin Vaidya <[email protected]>

* Apply PR comments

Signed-off-by: Ashwin Vaidya <[email protected]>

* rename dir

Signed-off-by: Ashwin Vaidya <[email protected]>

* delete llm_ollama

Signed-off-by: Ashwin Vaidya <[email protected]>

* Add ChatGPT

Signed-off-by: Ashwin Vaidya <[email protected]>

* Add ChatGPT

Signed-off-by: Ashwin Vaidya <[email protected]>

* Remove LLM model

Signed-off-by: Ashwin Vaidya <[email protected]>

---------

Signed-off-by: Ashwin Vaidya <[email protected]>

Author: ashwinvaidya17

pyproject.toml

@@ -56,7 +56,7 @@ core = [
     "open-clip-torch>=2.23.0,<2.26.1",
 ]
 openvino = ["openvino>=2024.0", "nncf>=2.10.0", "onnx>=1.16.0"]
-vlm = ["ollama", "transformers"]
+vlm = ["ollama", "openai", "transformers"]
 loggers = [
     "comet-ml>=3.31.7",
     "gradio>=4",

src/anomalib/models/__init__.py

@@ -26,8 +26,6 @@
     Ganomaly,
     Llava,
     Llavanext,
-    Llm,
-    Llmollama,
     Padim,
     Patchcore,
     ReverseDistillation,
@@ -65,8 +63,6 @@ class UnknownModelError(ModuleNotFoundError):
     "AiVad",
     "VlmAd",
     "WinClip",
-    "Llm",
-    "Llmollama",
     "Llava",
     "Llavanext",
 ]

src/anomalib/models/image/__init__.py

@@ -16,15 +16,13 @@
 from .ganomaly import Ganomaly
 from .llava import Llava
 from .llava_next import Llavanext
-from .llm import Llm
-from .llm_ollama import Llmollama
 from .padim import Padim
 from .patchcore import Patchcore
 from .reverse_distillation import ReverseDistillation
 from .rkde import Rkde
 from .stfpm import Stfpm
 from .uflow import Uflow
-from .vlm import VlmAd
+from .vlm_ad import VlmAd
 from .winclip import WinClip
 
 __all__ = [
@@ -47,8 +45,6 @@
     "Uflow",
     "VlmAd",
     "WinClip",
-    "Llm",
-    "Llmollama",
     "Llava",
     "Llavanext",
 ]