Update app.py

app.py

@@ -13,6 +13,17 @@ import os
 
 os.system("python3 -m pip install pytorch-lightning==1.7.0 aitextgen")
 
+with open("/usr/local/lib/python3.10/dist-packages/pytorch_lightning/callbacks/progress/rich_progress.py", "r") as file:
+    lines = file.readlines()
+
+    #replace line 20 "from torchmetrics.utilities.imports import _compare_version"
+    #with
+    #"from lightning_utilities.core.imports import compare_version as _compare_version"
+    lines[19] = "from lightning_utilities.core.imports import compare_version as _compare_version\n"
+
+with open("/usr/local/lib/python3.10/dist-packages/pytorch_lightning/callbacks/progress/rich_progress.py", "w") as file:
+    file.writelines(lines)
+
 import sys
 from jinja2 import Template
 from aitextgen import aitextgen