change app,py

app.py

@@ -5,33 +5,32 @@ 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(
+demo = gr.Interface(
         fn=greet, 
         inputs=[gr.Image(type="pil", label="Input"),
                 gr.Text(label="Prompt") ],
@@ -41,5 +40,7 @@ def main():
         article=article,
         examples=examples,
         css=css,
-        )
-    demo.launch(share=True)
+    )
+
+if __name__ == "__main__":
+    demo.launch()