no gpu at initialization

pages/summarization_playground.py

@@ -36,9 +36,10 @@ def generate_answer(sources, model, model_name, prompt):
     global __model_on_gpu__
 
     if __model_on_gpu__ != model_name:
-        model[__model_on_gpu__].cpu()
-        gc.collect()
-        torch.cuda.empty_cache()
+        if not __model_on_gpu__:
+            model[__model_on_gpu__].cpu()
+            gc.collect()
+            torch.cuda.empty_cache()
 
         model[model_name].cuda()
         __model_on_gpu__ = model_name