fix lightning

README.md

@@ -40,7 +40,7 @@ consider deploy on:
 > docs: [code](https://lightning.ai/docs/app/stable/levels/basic/real_lightning_component_implementations.html), [deploy](https://lightning.ai/docs/app/stable/workflows/run_app_on_cloud/)
 
 - requirements: 
-    - pip packages are listed in `requirements.txt`, note that some requirements are different from those in huggingface, and you need to modify some lines in `requirements.txt` according to the comment in the `requirements.txt`
+    - pip packages are listed in `requirements_lightning.txt`, because some requirements are different from those in huggingface. Rename it to `requirements.txt`
     - other pip packages, system packages and some big model weight files download commands, can be listed using a custom build config. Checkout `class CustomBuildConfig(BuildConfig)` in `app.py`. In a custom build config you can use many linux commands such as `wget` and `sudo apt-get update`. The custom build config will be executed on the `__init__()` of the `PythonServer` class
 - `init()` and `predict()` function: use `app.py`, implement the `PythonServer` class. Note: 
     - some packages haven't been installed when the file is called(these packages may be installed when `__init__()` is called), so some import code should be in the function, not at the top of the file, or you may get import errors.

app.py

@@ -43,7 +43,7 @@ class CustomBuildConfig(BuildConfig):
         cmd_31 = "sudo apt-get update"
         cmd3 = "sudo apt-get install libgl1-mesa-glx"
         cmd4 = "sudo apt-get install libglib2.0-0"
-        return [download_cmd, down2, down3, down4, cmd1, cmd2, cmd_31, cmd3, cmd4]
+        return [download_cmd, down2, down3, down4, cmd_31, cmd3, cmd4]
 
 
 class PyTorchServer(PythonServer):

requirements.txt

@@ -1,3 +1,4 @@
+# this one is for lightning.ai
 blendmodes==2022
 transformers==4.25.1
 accelerate==0.12.0
@@ -7,15 +8,11 @@ gradio==3.16.2
 numpy==1.23.3
 Pillow==9.4.0
 realesrgan==0.3.0
-# !!!!option1: if you use huggingface, you should use the following lines
-# !!!!option2: if you use lightning ai, you should comment the following lines
-lightning-api-access
-# torch==1.13.1+cu117
-# --extra-index-url https://download.pytorch.org/whl/cu117
-# torchvision==0.14.1+cu117
-# --extra-index-url https://download.pytorch.org/whl/cu117
+#lightning-api-access
 omegaconf==2.2.3
+requests>=2.26.0
 pytorch_lightning==1.7.6
+lightning-api-access
 scikit-image==0.19.2
 fonts
 font-roboto
@@ -23,7 +20,7 @@ timm==0.6.7
 piexif==1.1.3
 einops==0.4.1
 jsonmerge==1.8.0
-clean-fid==0.1.29
+clean-fid
 resize-right==0.0.2
 torchdiffeq==0.2.3
 kornia==0.6.7
@@ -33,6 +30,6 @@ GitPython==3.1.27
 torchsde==0.2.5
 safetensors==0.2.7
 httpcore<=0.15
-fastapi==0.90.1
+fastapi==0.88.0
 git+https://github.com/mlfoundations/open_clip.git@bb6e834e9c70d9c27d0dc3ecedeebeaeb1ffad6b
 git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1

requirements_hugging.txt

@@ -0,0 +1,38 @@
+# this one is for huggingface
+blendmodes==2022
+transformers==4.25.1
+accelerate==0.12.0
+basicsr==1.4.2
+gfpgan==1.3.8
+gradio==3.16.2
+numpy==1.23.3
+Pillow==9.4.0
+realesrgan==0.3.0
+# !!!!option1: if you use huggingface, you should use the following lines
+# !!!!option2: if you use lightning ai, you should comment the following lines
+torch==1.13.1+cu117
+--extra-index-url https://download.pytorch.org/whl/cu117
+torchvision==0.14.1+cu117
+--extra-index-url https://download.pytorch.org/whl/cu117
+omegaconf==2.2.3
+pytorch_lightning==1.7.6
+scikit-image==0.19.2
+fonts
+font-roboto
+timm==0.6.7
+piexif==1.1.3
+einops==0.4.1
+jsonmerge==1.8.0
+clean-fid==0.1.29
+resize-right==0.0.2
+torchdiffeq==0.2.3
+kornia==0.6.7
+lark==1.1.2
+inflection==0.5.1
+GitPython==3.1.27
+torchsde==0.2.5
+safetensors==0.2.7
+httpcore<=0.15
+fastapi==0.90.1
+git+https://github.com/mlfoundations/open_clip.git@bb6e834e9c70d9c27d0dc3ecedeebeaeb1ffad6b
+git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1