summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/DeepInfra.py
diff options
context:
space:
mode:
authornullstreak <139914347+nullstreak@users.noreply.github.com>2024-01-30 04:15:25 +0100
committerGitHub <noreply@github.com>2024-01-30 04:15:25 +0100
commit40ceb1c0e85d73ecc81e9552bc2ab4fb4928e5c6 (patch)
tree23af70968e2d4cceeaf503f216405e5cbf987042 /g4f/Provider/DeepInfra.py
parentMerge pull request #1527 from hlohaus/readme (diff)
downloadgpt4free-40ceb1c0e85d73ecc81e9552bc2ab4fb4928e5c6.tar
gpt4free-40ceb1c0e85d73ecc81e9552bc2ab4fb4928e5c6.tar.gz
gpt4free-40ceb1c0e85d73ecc81e9552bc2ab4fb4928e5c6.tar.bz2
gpt4free-40ceb1c0e85d73ecc81e9552bc2ab4fb4928e5c6.tar.lz
gpt4free-40ceb1c0e85d73ecc81e9552bc2ab4fb4928e5c6.tar.xz
gpt4free-40ceb1c0e85d73ecc81e9552bc2ab4fb4928e5c6.tar.zst
gpt4free-40ceb1c0e85d73ecc81e9552bc2ab4fb4928e5c6.zip
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