2

app.py

@@ -18,19 +18,19 @@ def color(image):
     result = output[OutputKeys.OUTPUT_IMG].astype(np.uint8)
     # result = result[...,::-1]
     # Generate a unique filename using UUID
-    unique_imgfilename = str(uuid.uuid4()) + '.png'
+    unique_imgfilename = 'colorize_'+str(uuid.uuid4()) + '.png'
     cv2.imwrite(unique_imgfilename, result)
     print('infer finished!')
     return (image, unique_imgfilename) , unique_imgfilename
     
 
-title = "Colorize Black and White Photos Online"
+title = "Colorize Black and White Photos"
 description = "Use AI image coloring algorithm to easily colorize your old black and white photos instead of traditional color filters."
 slider = ImageSlider(position=0.5,label='Original vs AI Processed')
 download_button = gr.File(label="Download")
 examples = [['./input.jpg'],]
 
-demo = gr.Interface(fn=color,inputs="image",outputs=[slider,download_button],examples=examples,title=title,description=description)
+demo = gr.Interface(fn=color,inputs="image",outputs=[slider,download_button],examples=examples,title=title,description='')
 
 if __name__ == "__main__":
-    demo.launch(share=False)
+    demo.launch(show_error=True)