1 files changed, 3 insertions, 3 deletions

diff --git a/g4f/__init__.py b/g4f/__init__.py
index 09b24b55..e5d3d4bf 100644
--- a/g4f/__init__.py
+++ b/g4f/__init__.py
@@ -1,12 +1,12 @@
 import sys
 from . import Provider
-from g4f.models import Model, ModelUtils
+from g4f import models
 
 logging = False
 
 class ChatCompletion:
     @staticmethod
-    def create(model: Model.model or str, messages: list, provider: Provider.Provider = None, stream: bool = False, auth: str = False, **kwargs):
+    def create(model: models.Model | str, messages: list, provider: Provider.Provider = None, stream: bool = False, auth: str = False, **kwargs):
         kwargs['auth'] = auth
         if provider and provider.working == False:
             return f'{provider.__name__} is not working'
@@ -19,7 +19,7 @@ class ChatCompletion:
         try:
             if isinstance(model, str):
                 try:
-                    model = ModelUtils.convert[model]
+                    model = models.ModelUtils.convert[model]
                 except KeyError:
                     raise Exception(f'The model: {model} does not exist')