Update app.py

app.py

@@ -85,25 +85,32 @@ def reload_models_task(model_choices):
         
         # Replace old models with new ones atomically
         with MODEL_LOCK:
-            # Clear old models from memory
-            for model_name in LOADED_MODELS:
-                if model_name in LOADED_MODELS:
-                    try:
-                        del LOADED_MODELS[model_name]
-                    except:
-                        pass
-                if model_name in LOADED_TOKENIZERS:
-                    try:
-                        del LOADED_TOKENIZERS[model_name]
-                    except:
-                        pass
+            # Store old models for cleanup
+            old_models = LOADED_MODELS.copy()
+            old_tokenizers = LOADED_TOKENIZERS.copy()
             
-            # Clear memory
-            clear_memory()
+            # Clear the dictionaries
+            LOADED_MODELS.clear()
+            LOADED_TOKENIZERS.clear()
             
             # Update with new models
             LOADED_MODELS.update(new_models)
             LOADED_TOKENIZERS.update(new_tokenizers)
+            
+            # Delete old model references
+            for model_name in old_models:
+                try:
+                    del old_models[model_name]
+                except:
+                    pass
+            for tokenizer_name in old_tokenizers:
+                try:
+                    del old_tokenizers[tokenizer_name]
+                except:
+                    pass
+            
+            # Clear memory
+            clear_memory()
         
         print(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] Model reload completed")