Initial commit.

.idea/workspace.xml

@@ -51,7 +51,7 @@
       <option name="number" value="Default" />
       <option name="presentableId" value="Default" />
       <updated>1647350746642</updated>
-      <workItem from="1647350750956" duration="7343000" />
+      <workItem from="1647350750956" duration="7452000" />
     </task>
     <task id="LOCAL-00001" summary="Initial commit.">
       <created>1647352693910</created>
@@ -186,7 +186,14 @@
       <option name="project" value="LOCAL" />
       <updated>1647357507081</updated>
     </task>
-    <option name="localTasksCounter" value="20" />
+    <task id="LOCAL-00020" summary="Initial commit.">
+      <created>1647358316467</created>
+      <option name="number" value="00020" />
+      <option name="presentableId" value="LOCAL-00020" />
+      <option name="project" value="LOCAL" />
+      <updated>1647358316467</updated>
+    </task>
+    <option name="localTasksCounter" value="21" />
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">

app.py

@@ -1,5 +1,3 @@
-import os
-
 import gradio as gr
 import numpy as np
 import requests
@@ -50,7 +48,7 @@ def download_file_from_google_drive(destination=WEIGHTS):
 
 def download_weights():
     # if not os.path.exists(WEIGHTS):
-    url = 'https://data.ciirc.cvut.cz/public/projects/2022DriveAndSegment/checkpoint_train_ep0_21.75miouHun.pth'
+    url = 'https://data.ciirc.cvut.cz/public/projects/2022DriveAndSegment/segmenter.pth'
     import urllib.request
     urllib.request.urlretrieve(url, WEIGHTS)
 
@@ -124,7 +122,7 @@ def get_transformations():
         transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])])
 
 
-download_file_from_google_drive()
+download_weights()
 model, window_size, window_stride = create_model()