summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/DeepInfra.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--g4f/Provider/DeepInfra.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/g4f/Provider/DeepInfra.py b/g4f/Provider/DeepInfra.py
index 09b9464e..8e36128d 100644
--- 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