Spaces:

Gradio-Blocks
/

protGPT2_gradioFold

Runtime error

simonduerr commited on May 24, 2022

Commit

9d493bd

1 Parent(s): 7649ab1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

         num_return_sequences=5,
         eos_token_id=0,
     )
+    torch.cuda.empty_cache()
     return sequences