added transformers.utils.move_cache()

app.py

@@ -13,6 +13,8 @@ import gradio as gr
 from torchvision.transforms import Compose, Resize, CenterCrop, ToTensor, Normalize
 from transformers import pipeline, GPT2LMHeadModel, GPT2Tokenizer
 
+transformers.utils.move_cache()
+
 example_1 = "ninja turtles fighting against a mosquito, in the sea"
 example_2 = "warrior fighting zombies with a sword, in the forest"
 example_3 = "western cowboy fighting against a dragon, in the desert"