delete to change models

pages/summarization_playground.py

@@ -11,7 +11,7 @@ import torch
 load_dotenv()
 
 __model_on_gpu__ = Model.__model_list__[0]
-model = {model_name: Model(model_name) for model_name in Model.__model_list__}
+model = {model_name: None for model_name in Model.__model_list__}
 
 random_label = '🔀 Random dialogue from dataset'
 examples = {
@@ -37,11 +37,11 @@ def generate_answer(sources, model, model_name, prompt):
 
     if __model_on_gpu__ != model_name:
         if not __model_on_gpu__:
-            model[__model_on_gpu__].cpu()
+            del model[__model_on_gpu__]
             gc.collect()
             torch.cuda.empty_cache()
 
-        model[model_name].cuda()
+        model[model_name] = Model(model_name)
         __model_on_gpu__ = model_name
 
     answer = model[model_name].gen(content)