Update app.py

app.py

@@ -210,6 +210,8 @@ class options:
         self.no_trace=no_trace
 
 def get_output(input_image):
+    ### Numpy -> PIL
+    input_image = Image.fromarray(input_image).convert('RGB')
     input_image.save(f"{BASE_DIR}/input/image.jpg")
     source = f"{BASE_DIR}/input"
     opt = options(weights='logo_detection.pt',source=source)
@@ -222,5 +224,5 @@ def get_output(input_image):
         return input_image
 
 
-demo = gr.Interface(fn=get_output, type='pil', inputs="image", outputs="image")
+demo = gr.Interface(fn=get_output, inputs="image", outputs="image")
 demo.launch(debug=True)