Fix DeepInfra: Model is not supported

5 months ago · 40ceb1c0e8
parent 280ddf57b0
commit 40ceb1c0e8
1 changed files with 2 additions and 1 deletions
--- a/g4f/Provider/DeepInfra.py
+++ b/g4f/Provider/DeepInfra.py
@ -17,7 +17,8 @@ class DeepInfra(AsyncGeneratorProvider, ProviderModelMixin):
    def get_models(cls):
        if not cls.models:
            url = 'https://api.deepinfra.com/models/featured'
-            cls.models = requests.get(url).json()
+            models = requests.get(url).json()
+            cls.models = [model['model_name'] for model in models]
        return cls.models

    @classmethod