inference

app.py

@@ -1,51 +1,45 @@
 import gradio as gr
 from inference import infer
 
-import os
-
-# setup.sh 파일 실행
-os.system("install.sh")
-
-
 def greet(image, prompt):
     restore_img = infer(img=image, text_prompt=prompt)
     return restore_img
 
+def main():
+    title = "🖼️ ICDR 🖼️"
+    description = ''' ## ICDR: Image Restoration Framework for Composite Degradation following Human Instructions
+    Our Github : https://github.com/
 
-title = "🖼️ ICDR 🖼️"
-description = ''' ## ICDR: Image Restoration Framework for Composite Degradation following Human Instructions
-Our Github : https://github.com/
-
-Siwon Kim, Donghyeon Yoon
+    Siwon Kim, Donghyeon Yoon
 
-Ajou Univ
-'''
+    Ajou Univ
+    '''
 
 
-article = "<p style='text-align: center'><a href='https://github.com/' target='_blank'>ICDR</a></p>"
+    article = "<p style='text-align: center'><a href='https://github.com/' target='_blank'>ICDR</a></p>"
 
-#### Image,Prompts examples
-examples = [['input/00010.png', "I love this photo, could you remove the haze and more brighter?"],
-            ['input/00058.png', "I have to post an emotional shot on Instagram, but it was shot too foggy and too dark. Change it like a sunny day and brighten it up!"]]
+    #### Image,Prompts examples
+    examples = [['input/00010.png', "I love this photo, could you remove the haze and more brighter?"],
+                ['input/00058.png', "I have to post an emotional shot on Instagram, but it was shot too foggy and too dark. Change it like a sunny day and brighten it up!"]]
 
-css = """
-    .image-frame img, .image-container img {
-        width: auto;
-        height: auto;
-        max-width: none;
-    }
-"""
+    css = """
+        .image-frame img, .image-container img {
+            width: auto;
+            height: auto;
+            max-width: none;
+        }
+    """
 
 
-demo = gr.Interface(
-    fn=greet, 
-    inputs=[gr.Image(type="pil", label="Input"),
-            gr.Text(label="Prompt") ],
-    outputs=[gr.Image(type="pil", label="Ouput")],
-    title=title,
-    description=description,
-    article=article,
-    examples=examples,
-    css=css,
-    )
-demo.launch(share=True)
+    demo = gr.Interface(
+        fn=greet, 
+        inputs=[gr.Image(type="pil", label="Input"),
+                gr.Text(label="Prompt") ],
+        outputs=[gr.Image(type="pil", label="Ouput")],
+        title=title,
+        description=description,
+        article=article,
+        examples=examples,
+        css=css,
+        )
+    demo.launch(share=True)

inference.ipynb

@@ -0,0 +1,33 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install mmcv==2.2.0 -f https://download.openmmlab.com/mmcv/dist/cu121/torch2.4/index.html\n",
+    "!pip install git+https://github.com/openai/CLIP.git\n",
+    "!pip install scikit-video"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from app.py import main\n",
+    "\n",
+    "main()"
+   ]
+  }
+ ],
+ "metadata": {
+  "language_info": {
+   "name": "python"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

requirements.txt

@@ -7,4 +7,5 @@ scikit-image
 scikit-video
 matplotlib
 tqdm
-git+https://github.com/openai/CLIP.git
+git+https://github.com/openai/CLIP.git
+mmcv-lite