Spaces:
Sleeping
Sleeping
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() |