Adding that ModelDetails is a dataclass.

app.py

@@ -1,5 +1,6 @@
 import os
 import datetime
+from dataclasses import dataclass
 
 import gradio as gr
 from huggingface_hub import HfApi 
@@ -16,6 +17,7 @@ API = HfApi(token=TOKEN)
 
 
 ## All the model information that we might need
+@dataclass
 class ModelDetails:
     name: str
     display_name: str = ""