- app.py +6 -0
- install.sh +2 -0
- requirements.txt +1 -2
app.py
CHANGED
|
@@ -1,6 +1,12 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
from inference import infer
|
| 3 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
def greet(image, prompt):
|
| 5 |
restore_img = infer(img=image, text_prompt=prompt)
|
| 6 |
return restore_img
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
from inference import infer
|
| 3 |
|
| 4 |
+
import os
|
| 5 |
+
|
| 6 |
+
# setup.sh 파일 실행
|
| 7 |
+
os.system("install.sh")
|
| 8 |
+
|
| 9 |
+
|
| 10 |
def greet(image, prompt):
|
| 11 |
restore_img = infer(img=image, text_prompt=prompt)
|
| 12 |
return restore_img
|
install.sh
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
pip install --upgrade pip
|
| 2 |
+
pip install mmcv==2.2.0 -f https://download.openmmlab.com/mmcv/dist/cu121/torch2.0/index.html
|
requirements.txt
CHANGED
|
@@ -7,5 +7,4 @@ scikit-image
|
|
| 7 |
scikit-video
|
| 8 |
matplotlib
|
| 9 |
tqdm
|
| 10 |
-
git+https://github.com/openai/CLIP.git
|
| 11 |
-
mmcv==2.2.0
|
|
|
|
| 7 |
scikit-video
|
| 8 |
matplotlib
|
| 9 |
tqdm
|
| 10 |
+
git+https://github.com/openai/CLIP.git
|
|
|