|
import os |
|
import pandas as pd |
|
|
|
class CachedModels: |
|
def __init__(self): |
|
csv_url = "https://docs.google.com/spreadsheets/d/1tAUaQrEHYgRsm1Lvrnj14HFHDwJWl0Bd9x0QePewNco/export?format=csv&gid=1977693859" |
|
if os.path.exists("spreadsheet.csv"): |
|
self.cached_data = pd.read_csv("spreadsheet.csv") |
|
else: |
|
self.cached_data = pd.read_csv(csv_url) |
|
self.cached_data.to_csv("spreadsheet.csv", index=False) |
|
|
|
self.models = {} |
|
for _, row in self.cached_data.iterrows(): |
|
filename = row['Filename'] |
|
url = None |
|
for value in row.values: |
|
if isinstance(value, str) and "huggingface" in value: |
|
url = value |
|
break |
|
if url: |
|
self.models[filename] = url |
|
|
|
def get_models(self): |
|
return self.models |