update

app.py

@@ -5,7 +5,7 @@ from aitextgen import aitextgen
 def text_gen(text):
 
     model_name = "EleutherAI/gpt-neo-125M"
-    model = aitextgen(model = model_name, to_gpu = True)
+    model = aitextgen(model = model_name)
     text_generated = model.generate_one(max_length=1000, prompt = text, no_repeat_ngram_size = 3)
 
     return text_generated