Update README with YAML config

README.md

@@ -1,2 +1,11 @@
-# deoldify_colorizer
-Colorizer is a Gradio program based on Deoldify (https://github.com/jantic/DeOldify)
+---
+title: Colorize
+emoji: 🌟
+colorFrom: blue
+colorTo: purple
+sdk: gradio
+sdk_version: 3.4
+app_file: app.py
+pinned: false
+license: mit
+---

app.py

@@ -7,13 +7,13 @@ from datetime import datetime
 
 model = hub.Module(name='deoldify')
 # NOTE:  Max is 45 with 11GB video cards. 35 is a good default
-render_factor=35
+render_factor=30
 
 
 def colorize_image(image):
     # now = datetime.now().strftime("%Y%m%d-%H%M%S-%f")
-    # if not os.path.exists("./output"):
-    #     os.makedirs("./output")
+    if not os.path.exists("./output"):
+        os.makedirs("./output")
     # if image is not None:
     #    image.convert("RGB").save(f"./output/{now}-input.jpg")    
     model.predict(image.name)
@@ -35,4 +35,4 @@ def run_code():
     create_interface()
 
 # The main function
-run_code()
+run_code()