Initial commit.

app.py

@@ -58,6 +58,7 @@ def download_file_from_google_drive(destination=WEIGHTS):
 
 
 def download_weights():
+    print('Downloading weights...')
     # if not os.path.exists(WEIGHTS):
     url = 'https://data.ciirc.cvut.cz/public/projects/2022DriveAndSegment/segmenter_nusc.pth'
     import urllib.request
@@ -223,6 +224,8 @@ examples = [  # 'examples/img5.jpeg',
     'examples/snow1.jpg']
 examples += ['examples/cs{}.jpg'.format(i) for i in range(2, 5)]
 
+examples = ['examples/cs2.jpg']
+
 # predict(examples[0])
 
 iface = gr.Interface(predict, inputs=gr.Image(type='filepath'), title=title, description=description,