dodd869 commited on
Commit
4c03b44
·
verified ·
1 Parent(s): a24e8c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -14
app.py CHANGED
@@ -3,20 +3,9 @@ import gradio as gr
3
  from huggingface_hub import InferenceClient
4
  import requests
5
 
6
- def fetch_models():
7
- models = []
8
- page = 1
9
- while True:
10
- url = f"https://huggingface.co/models-json?num_parameters=min%3A32B&inference_provider=novita&p={page}&sort=modified&withCount=true"
11
- res = requests.get(url).json()
12
- page_models = [m['id'] for m in res.get('models', [])]
13
- if not page_models:
14
- break
15
- models.extend(page_models)
16
- page += 1
17
- return models
18
-
19
- model_ids = fetch_models()
20
 
21
  client = InferenceClient(provider="novita", api_key=os.environ["HF_TOKEN"])
22
 
 
3
  from huggingface_hub import InferenceClient
4
  import requests
5
 
6
+ url = "https://huggingface.co/models-json?num_parameters=min%3A64B&inference_provider=novita&sort=modified&withCount=true"
7
+ res = requests.get(url).json()
8
+ model_ids = [m['id'] for m in res.get('models', [])]
 
 
 
 
 
 
 
 
 
 
 
9
 
10
  client = InferenceClient(provider="novita", api_key=os.environ["HF_TOKEN"])
11