Fix errors in demo.

app.py

@@ -3,9 +3,12 @@ import yolov5
 import numpy as np
 from PIL import Image, ImageDraw, ImageFilter
 from transformers import OneFormerProcessor, OneFormerForUniversalSegmentation
+import torchvision.transforms
+import torch
 
 person_processor = OneFormerProcessor.from_pretrained("shi-labs/oneformer_cityscapes_swin_large")
 person_model = OneFormerForUniversalSegmentation.from_pretrained("shi-labs/oneformer_cityscapes_swin_large")
+transform = torchvision.transforms.ToPILImage()
 
 lp_model = yolov5.load('keremberke/yolov5m-license-plate')  
 lp_model.conf = 0.25  # NMS confidence threshold
@@ -68,5 +71,6 @@ def test_gradio(image):
     anonymized = Image.composite(image, blurred, mask)
     return anonymized
 
-interface = gr.Interface(fn=test_gradio, inputs=gr.Image(type="pil"), outputs="image")
-interface.launch(share=True)
+demo = gr.Interface(fn=test_gradio, inputs=gr.Image(type="pil"), outputs=gr.Image(type="pil"))
+demo.launch()
+#demo.launch(server_name="localhost", server_port=8080)

requirements.txt

@@ -1,4 +1,6 @@
 yolov5
 transformers
 Pillow
-numpy
+numpy
+torch
+torchvision