renamed TextToImage

deepinfra/models/base/__init__.py

@@ -1,4 +1,4 @@
 from .base import BaseModel
-from .image_generation import ImageGeneration
+from .text_to_image import TextToImage
 from .embeddings import Embeddings
 from .automatic_speech_recognition import AutomaticSpeechRecognition

deepinfra/models/base/automatic_speech_recognition.py

@@ -11,13 +11,15 @@
 class AutomaticSpeechRecognition(BaseModel):
 
     """
-    Initializes the automatic speech recognition model.
-    @param endpoint: The endpoint of the model or the model name.
-    @param auth_token: The API key to authenticate the requests.
     @docs Check the available models at https://deepinfra.com/models/automatic-speech-recognition
     """
 
     def __init__(self, endpoint: str, auth_token: str = None):
+        """
+        Initializes the automatic speech recognition model.
+        @param endpoint: The endpoint of the model or the model name.
+        @param auth_token: The API key to authenticate the requests.
+        """
         super().__init__(endpoint, auth_token)
 
     def generate(self, body) -> AutomaticSpeechRecognitionResponse:

deepinfra/models/base/image_generation.py → deepinfra/models/base/text_to_image.py

@@ -1,10 +1,11 @@
 from deepinfra.models import BaseModel
 
 
-class ImageGeneration(BaseModel):
+class TextToImage(BaseModel):
 
     """
     Initializes one of the DeepInfra image generation models.
+    @docs Check the available models at https://deepinfra.com/models/text-to-image
     """
 
     def __init__(self, endpoint: str, auth_token: str):