Update app.py

app.py

@@ -9,6 +9,7 @@ import matplotlib
 matplotlib.use("Agg")
 import matplotlib.pyplot as plt
 from transformers import pipeline as pl
+from GPUtil import showUtilization as gpu_usage
 
 import pandas as pd
 import numpy as np
@@ -16,7 +17,7 @@ import matplotlib.pyplot as plt
 import sys
 import plotly.graph_objects as go
 import torch
-
+import gc
 print('GPU available',torch.cuda.is_available())
 print('__CUDA Device Name:',torch.cuda.get_device_name(0))
 print(os.getcwd())
@@ -90,12 +91,16 @@ def run_protgpt2(startsequence, length, repetitionPenalty, top_k_poolsize, max_s
         num_return_sequences=max_seqs,
         eos_token_id=0,
     )
+    print(gpu_usage())
     del protgpt2
     torch.cuda.empty_cache()
+    gc.collect()
+    print(gpu_usage())
     return sequences
 
 
 def run_alphafold(startsequence):
+    print(gpu_usage())
     model_runners = {}
     models = ["model_1"]  # ,"model_2","model_3","model_4","model_5"]
     for model_name in models:
@@ -116,6 +121,9 @@ def run_alphafold(startsequence):
         **mk_mock_template(query_sequence),
     }
     plddts = predict_structure("test", feature_dict, model_runners)
+    del model_runners
+    gc.collect()
+    print(gpu_usage())
     return plddts["model_1"]