model-explorer / app.py
Omnibus's picture
Update app.py
93e1bf4
raw
history blame
2.39 kB
import gradio as gr
import json
from huggingface_hub import HfApi, ModelFilter, list_models
api = HfApi()
def model_explorer():
# List all models
filt = ModelFilter(task="text-classification")
this = api.list_models(limit=100,filter=filt,cardData=True)
for i,mod in enumerate(this):
print(dir(mod))
print(mod.id)
print(mod.downloads)
print(mod.likes)
print(mod.pipeline_tag)
out = mod.id
return out
def tasks_json():
with open("tasks.json", "r") as read_file:
print("Read JSON file")
tasks = json.load(read_file)
print("Before Pretty Printing JSON Data")
print(tasks)
def fix_json():
new_json=[]
with open("tasks.json", "r") as fix_file:
for i, line in enumerate(fix_file.readlines()):
#print (line)
if "name:" in line:
out = line.replace("name:",'"name":')
new_json.append(out)
print (out)
elif "modality:" in line:
out = line.replace("modality:",'"modality":')
new_json.append(out)
print (out)
elif "type:" in line:
out = line.replace("type:",'"type":')
new_json.append(out)
print (out)
elif "subtasks:" in line:
out = line.replace("subtasks:",'"subtasks":')
new_json.append(out)
print (out)
elif "color:" in line:
out = line.replace("color:",'"color":')
new_json.append(out)
print (out)
elif "hideInModels:" in line:
out = line.replace("hideInModels:",'"hideInModels":')
new_json.append(out)
print (out)
elif "hideInDatasets:" in line:
out = line.replace("hideInDatasets:",'"hideInDatasets":')
new_json.append(out)
print (out)
else:
new_json.append(line)
print(line)
#print (json.dumps(new_json))
#new_out=new_json.decode("utf-8")
for i,line in enumerate(new_json):
try:
line.decode("utf-8")
except Exception:
pass
if line.strip():
print (line)
fix_json()