Update model_registry.py

model_registry.py

@@ -12,10 +12,15 @@ LIST OF WEBSITES TO REFER - {', '.join(websites)}
 IMPORTANT NOTE - You must always include citations source URLs in the response for user transparency. Provide information that is relevant and up-to-date as of this {current_date} date.
 """
 
+supported_models = {
+    'llama3.1-8B': 'llama-3.1-sonar-small-128k-online',
+    'llama3.1-70B': 'llama-3.1-sonar-large-128k-online',
+    'llama3.1-405B': 'llama-3.1-sonar-huge-128k-online'
+}
 
 model_card = {
   "perplexity": {
-    "model": "llama-3.1-sonar-huge-128k-online", #"llama-3.1-sonar-large-128k-online",
+    "model": supported_models['llama3.1-405B'],
     "url": "https://api.perplexity.ai/chat/completions",
     "inference_config": {
       "max_tokens": 4094,