Update app.py

app.py

@@ -32,6 +32,9 @@ def load_model(model_name):
     if model_name == current_model_name:
         return current_model
 
+    # Load the repo info related to the selected model from the available models dictionary
+    info = AVAILABLE_MODELS.get(model_name)
+    
     model_path = hf_hub_download(
         repo_id=info["repo_id"],
         filename=info["filename"],