Update app.py

app.py

@@ -61,10 +61,9 @@ class ModelManager:
                     with open(cache_file, "rb") as f:
                         self.models[model_name] = pickle.load(f)
                 else:
-                    llama_model = Llama.from_pretrained(repo_id=model_config['repo_id'], filename=model_config['filename'], use_auth_token=HUGGINGFACE_TOKEN)
-                    self.models[model_name] = llama_model
+                    self.models[model_name] = Llama.from_pretrained(repo_id=model_config['repo_id'], filename=model_config['filename'], use_auth_token=HUGGINGFACE_TOKEN)
                     with open(cache_file, "wb") as f:
-                        pickle.dump(llama_model, f)
+                        pickle.dump(self.models[model_name], f)
             except Exception as e:
                 print(f"Error loading {model_name}: {e}")
                 self.models[model_name] = None