Update app.py

app.py

@@ -18,6 +18,7 @@ import sys
 import plotly.graph_objects as go
 import torch
 import gc
+from numba import cuda
 print('GPU available',torch.cuda.is_available())
 print('__CUDA Device Name:',torch.cuda.get_device_name(0))
 print(os.getcwd())
@@ -97,6 +98,8 @@ def run_protgpt2(startsequence, length, repetitionPenalty, top_k_poolsize, max_s
     del protgpt2
     gc.collect()
     torch.cuda.empty_cache()
+    device = cuda.get_current_device()
+    device.reset()
     print(gpu_usage())
     return sequences