Update app.py

app.py

@@ -1,4 +1,3 @@
-# Import necessary libraries
 import gradio as gr
 import sys
 from huggingface_hub import ModelCard, HfApi
@@ -24,19 +23,19 @@ MODEL_ID = "mlabonne/NeuralBeagle14-7B"
 
 # Define a class to cache model cards
 class CachedModelCard(ModelCard):
-  _cache = {}
-    
-  @classmethod
-  def load(cls, model_id: str, **kwargs) -> "ModelCard":
-    if model_id not in cls._cache:
-      try:
-        print('REQUEST ModelCard:', model_id)
-        cls._cache[model_id] = super().load(model_id, **kwargs)
-      except:
-        cls._cache[model_id] = None
-    else:
-      print('CACHED:', model_id)
-    return cls._cache[model_id]
+    _cache = {}
+
+    @classmethod
+    def load(cls, model_id: str, **kwargs) -> "ModelCard":
+        if model_id not in cls._cache:
+            try:
+                print('REQUEST ModelCard:', model_id)
+                cls._cache[model_id] = super().load(model_id, **kwargs)
+            except:
+                cls._cache[model_id] = None
+        else:
+            print('CACHED:', model_id)
+        return cls._cache[model_id]
 
 # Function to get model names from a YAML file
 def get_model_names_from_yaml(url):
@@ -65,7 +64,7 @@ def get_license_color(model):
         print(f"Error retrieving license for {model}: {e}")
         return 'lightgray'
 
-        
+
 # Function to find model names in the family tree
 def get_model_names(model, genealogy, found_models=None, visited_models=None):
     print('---')
@@ -138,7 +137,7 @@ def calculate_width_at_depth(G, root):
             queue.append((child, depth + 1))
     return depth_count
 
-    
+
 # Function to create the family tree
 def create_family_tree(start_model):
     genealogy = defaultdict(list)
@@ -189,7 +188,7 @@ def create_family_tree(start_model):
     plt.legend(handles=legend_elements, loc='upper left')
 
     plt.title(f"{start_model}'s Family Tree", fontsize=20)
-    
+
     # Capture the plot as an image in memory
     img_buffer = BytesIO()
     plt.savefig(img_buffer, format='png', bbox_inches='tight')
@@ -200,7 +199,7 @@ def create_family_tree(start_model):
     img = Image.open(img_buffer)
 
     return img
-    
+
 with gr.Blocks() as demo:
     gr.Markdown(TITLE)
     model_id = gr.Textbox(label="Model ID", value="mlabonne/NeuralBeagle14-7B")
@@ -209,8 +208,6 @@ with gr.Blocks() as demo:
     btn.click(fn=create_family_tree, inputs=model_id, outputs=out)
 
 demo.queue(api_open=False).launch(show_api=False)
-    
-
-create_family_tree(MODEL_ID)
 
 
+create_family_tree(MODEL_ID)