Update app.py

app.py

@@ -92,11 +92,9 @@ def run_protgpt2(startsequence, length, repetitionPenalty, top_k_poolsize, max_s
         num_return_sequences=max_seqs,
         eos_token_id=0,
     )
+    print("Cleaning up after protGPT2")
     print(gpu_usage())
-    print(sequences)
-    print(type(sequences))
     del protgpt2
-    gc.collect()
     torch.cuda.empty_cache()
     device = cuda.get_current_device()
     device.reset()
@@ -126,8 +124,10 @@ def run_alphafold(startsequence):
         **mk_mock_template(query_sequence),
     }
     plddts = predict_structure("test", feature_dict, model_runners)
-    del model_runners
-    gc.collect()
+    print("Cleaning up after AF2")
+    print(gpu_usage())
+    device = cuda.get_current_device()
+    device.reset()
     print(gpu_usage())
     return plddts["model_1"]