Update app.py

app.py

@@ -13,6 +13,8 @@ import os
 
 os.system("python3 -m pip install pytorch-lightning==1.7.0 aitextgen")
 
+from aitextgen import aitextgen
+
 with open("/usr/local/lib/python3.10/dist-packages/pytorch_lightning/callbacks/progress/rich_progress.py", "r") as file:
     lines = file.readlines()