| 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 |