Update app.py

app.py

@@ -90,6 +90,7 @@ def run_protgpt2(startsequence, length):
         num_return_sequences=5,
         eos_token_id=0,
     )
+    torch.cuda.empty_cache()
     return sequences