ModelBase

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import requests
+import re
 from typing import Optional
 
 
@@ -48,8 +49,15 @@ def is_architecture_supported(architecture: str) -> Optional[bool]:
         response = requests.get(LLAMA_CPP_SOURCE)
         response.raise_for_status()
         content = response.text
-        models = [line for line in content.splitlines() if "@Model.register" in line]
-        return any(architecture in entry for entry in models)
+        pattern = r"@ModelBase\.register\((.*?)\)"
+        matches = re.findall(pattern, content, re.DOTALL)
+        model_names = [
+            name.strip()
+            for match in matches
+            for name in match.replace('\n', '').replace('"', '').split(',')
+            if name.strip()
+        ]
+        return any(architecture in entry for entry in model_names)
     except:
         return None